POJ 3368 Frequent values RMQ 训练指南 好题

#include<cstdio>
#include<cstring>

const int maxn=1e5+5;
const int inf=0x3f3f3f3f;

inline int max(int x,int y)
{
    return x>y?x:y;
}

int a[maxn];
int left[maxn];
int right[maxn];
int num[maxn];
int dp[maxn][20];

void init()
{
    memset(a,0,sizeof a);
    memset(left,-1,sizeof left);
    memset(right,-1,sizeof right);
}

void init_rmq(int n)
{
    for(int i=1;i<=n;i++)
        dp[i][0]=a[i];
    for(int j=1;(1<<j)<=n;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 query(int l,int r)
{
    if(l>r)
        return 0;
    int k=0;
    while((1<<(k+1))<=r-l+1)
        k++;
    return max(dp[l][k],dp[r-(1<<k)+1][k]);
}

void solve(int n,int q)
{
    init_rmq(n);
    for(int i=0;i<q;i++)
    {
        int L,R;
        scanf("%d%d",&L,&R);
        if(num[L]==num[R])
        {
            printf("%d
",R-L+1);
            continue;
        }
        int ans=max(right[num[L]]-L+1,R-left[num[R]]+1);
        ans=max(ans,query(num[L]+1,num[R]-1));
        printf("%d
",ans);
    }
    return ;
}

int main()
{
    int cnt;
    while(scanf("%d",&cnt))
    {
        if(!cnt)
            break;
        int q;
        scanf("%d",&q);
        init();
        int pre=-inf;
        int n=0;
        for(int i=1;i<=cnt;i++)
        {
            int u;
            scanf("%d",&u);
            if(u!=pre)
            {
                right[n]=i-1;
                a[++n]=1;
                left[n]=i;
            }
            else
                a[n]++;
            num[i]=n;
            pre=u;
        }
        right[n]=cnt;

        solve(n,q);
    }
    return 0;
}