POJ 3368 Frequent values（线段树/RMQ）

题目链接：
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;
}