POJ 3368 Frequent values(线段树/RMQ)

来源:互联网 发布:ab post json请求 编辑:程序博客网 时间:2024/06/05 20:56

题目链接:
POJ 3368 Frequent values
题意:
给一串非降序数字序列,然后有Q个查询,每次查询区间[a,b]内的最大连续相同数字长度,数组下标从1开始。
如数字序列为:-1 -1 1 2 2 2 ,区间[1,3]内的最大连续相同数字长度为2,而区间[3,6]内的答案为3.
分析:
①用线段树存储树中区间的最大连续相同区间长度。在建树和查询区间被左右儿子“瓜分”的时候,注意左儿子的右端点和右儿子的左端点相同的情况。
②RMQ.如果用RMQ解决的话,需要一步转化:记录以data[i]结尾的最大连续相同区间长度为num[i].那么在查询区间[a,b]的最大连续相同区间长度时,先计算一下以区间左端点为起点的最大连续相同区间长度,然后后面的区间内的最大连续相同区间长度就可以用RMQ解决,即求区间最大值。最后两者取最大。
例如区间[a,b]内的第一个大于data[a]的数据的下标为ind,那么以data[a]为起点的连续相同区间长度是ind-a,后面的区间是[ind,b],再求出这个区间的最大值,然后和ind-a比较两者取最大即可。
注意:后面的区间的起点是ind,而不是a!

//Segment Tree Solution 6948K 1391MS#include <cstdio>#include <cstring>#include <cmath>#include <algorithm>#define lson(cur) (cur<<1)#define rson(cur) ((cur<<1)|1)using namespace std;const int MAX_N=100010;int n,Q;int data[MAX_N];struct Segtree{    int left,right,len,num;    int left_num,right_num;}segtree[MAX_N<<2];inline void push_up(int cur){    int left=segtree[cur].left;    int right=segtree[cur].right;    int mid=(left+right)>>1;    segtree[cur].left_num=segtree[lson(cur)].left_num;    if(segtree[cur].left_num==segtree[lson(cur)].len&&data[mid]==data[mid+1]) segtree[cur].left_num+=segtree[rson(cur)].left_num;    segtree[cur].right_num=segtree[rson(cur)].right_num;    if(segtree[cur].right_num==segtree[rson(cur)].len&&data[mid]==data[mid+1]) segtree[cur].right_num+=segtree[lson(cur)].right_num;    int tmp=max(segtree[lson(cur)].num,segtree[rson(cur)].num);    if(data[mid]==data[mid+1]){        tmp=max(tmp,segtree[lson(cur)].right_num+segtree[rson(cur)].left_num);    }    segtree[cur].num=tmp;    //printf("cur=%d left=%d right=%d num=%d\n",cur,segtree[cur].left,segtree[cur].right,segtree[cur].num);}inline void build(int left,int right,int cur){    segtree[cur].left=left;    segtree[cur].right=right;    segtree[cur].len=right-left+1;    if(left==right){        segtree[cur].left_num=segtree[cur].right_num=segtree[cur].num=1;        //printf("cur=%d left=%d right=%d num=%d\n",cur,segtree[cur].left,segtree[cur].right,segtree[cur].num);        return;    }    int mid=(left+right)>>1;    build(left,mid,lson(cur));    build(mid+1,right,rson(cur));    push_up(cur);}inline int query(int a,int b,int cur){    int left=segtree[cur].left;    int right=segtree[cur].right;    if(left==a&&right==b){        return segtree[cur].num;    }    int mid=(left+right)>>1;    if(a>mid) return query(a,b,rson(cur));    else if(b<=mid) return query(a,b,lson(cur));    else {        int tmp=max(query(a,mid,lson(cur)),query(mid+1,b,rson(cur)));        if(data[mid]==data[mid+1]){            int ans_left=min(segtree[lson(cur)].right_num,mid-a+1);            int ans_right=min(segtree[rson(cur)].left_num,b-(mid+1)+1);            //printf("a=%d b=%d ans_left=%d ans_right=%d\n",a,b,ans_left,ans_right);            tmp=max(tmp,ans_left+ans_right);        }        return tmp;    }}int main(){    //freopen("poj3368in.txt","r",stdin);    while(~scanf("%d",&n)&&n){        scanf("%d",&Q);        for(int i=1;i<=n;i++){            scanf("%d",&data[i]);        }        build(1,n,1);        for(int i=0;i<Q;i++){            int a,b;            scanf("%d%d",&a,&b);            printf("%d\n",query(a,b,1));        }    }    return 0;}
//RMQ Solution 9004K 1266MS#include <cstdio>#include <cstring>#include <cmath>#include <algorithm>using namespace std;const int MAX_N=100010;int n,Q;int data[MAX_N],num[MAX_N],dp[MAX_N][20];//num[i]表示以data[i]结尾的最大连续相同区间长度//dp[i][j]表示以data[i]到以data[i+(1<<j)-1]为结尾的最大连续相同区间长度inline void RMQ(){    for(int i=1;i<=n;i++){        dp[i][0]=num[i];    }    int k=(int)(log2(n+1));    for(int j=1;j<=k;j++){        for(int i=1;i+(1<<j)-1<=n;i++){            dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);        }    }}int main(){    //freopen("poj3368in.txt","r",stdin);    while(~scanf("%d",&n)&&n){        scanf("%d",&Q);        for(int i=1;i<=n;i++){            scanf("%d",&data[i]);            if(i==1||data[i]!=data[i-1]){                num[i]=1;            }else {//data[i]==data[i-1]                num[i]=num[i-1]+1;            }        }        RMQ();        for(int i=0;i<Q;i++){            int a,b,ans;            scanf("%d%d",&a,&b);            //查询区间的左端点的最大连续相同区间长度            int ind=upper_bound(data+1,data+1+n,data[a])-data;//ind是输入中第一个大于data[a]的数据的下标            //printf("ind=%d\n",ind);            if(ind>b) ans=b-a+1;            else {                ans=ind-a;//ans是以区间左端点为起点的最大连续相同区间长度                int k=(int)(log2(b-ind+1));                ans=max(ans,max(dp[ind][k],dp[b-(1<<k)+1][k]));            }            printf("%d\n",ans);        }    }    return 0;}
0 0
原创粉丝点击