[BZOJ]3956: Count 主席树+单调栈

Solution

问题其实就是求两端点都为区间最大值的点对数量，先用单调栈求出每个点作为最大值可以延伸到的左右端点，然后可以发现答案都是这样的：对于一对$(x,y)$，要么是$x$向右延伸到$y-1$，要么是$y$向左延伸到$x+1$，所以问题就转化为区间内某个范围内的数有多少个，直接主席树即可。
注意相同的数的处理，详情可以看代码。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=300010;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,m,type,a[Maxn],sta[Maxn],top=0,L[Maxn],R[Maxn],mx[Maxn][19],Log[Maxn];
int query_mx(int l,int r)
{
    int t=Log[r-l+1];
    return max(mx[l][t],mx[r-(1<<t)+1][t]);
}
struct Tree
{
    int tot,root[Maxn],lc[Maxn*20],rc[Maxn*20],c[Maxn*20];
    void insert(int &u,int l,int r,int p)
    {
        if(!u)u=++tot;
        c[u]++;
        if(l==r)return;
        int mid=l+r>>1;
        if(p<=mid)insert(lc[u],l,mid,p);
        else insert(rc[u],mid+1,r,p);
    }
    void merge(int &u1,int u2)
    {
        if(!u1){u1=u2;return;}
        if(!u2)return;
        c[u1]+=c[u2];
        merge(lc[u1],lc[u2]),merge(rc[u1],rc[u2]);
    }
    int query(int r1,int r2,int l,int r,int fl,int fr)
    {
        if(fl>fr)return 0;
        if(!r2)return 0;
        if(l==fl&&r==fr)return c[r2]-c[r1];
        int mid=l+r>>1;
        if(fr<=mid)return query(lc[r1],lc[r2],l,mid,fl,fr);
        if(fl>mid)return query(rc[r1],rc[r2],mid+1,r,fl,fr);
        return query(lc[r1],lc[r2],l,mid,fl,mid)+query(rc[r1],rc[r2],mid+1,r,mid+1,fr);
    }
}T1,T2;
int main()
{
    n=read(),m=read(),type=read();
    Log[1]=0;for(int i=2;i<=n;i++)Log[i]=Log[i>>1]+1;
    for(int i=1;i<=n;i++)a[i]=mx[i][0]=read();
    for(int j=1;(1<<j)<=n;j++)
    for(int i=1;i+(1<<j)-1<=n;i++)
    mx[i][j]=max(mx[i][j-1],mx[i+(1<<(j-1))][j-1]);
    for(int i=1;i<=n;i++)
    {
        while(top&&a[i]>=a[sta[top]])R[sta[top--]]=i-1;
        if(!top)L[i]=1;else L[i]=sta[top]+1;
        sta[++top]=i;
    }
    while(top)R[sta[top--]]=n;
    for(int i=1;i<=n;i++)if(L[i]<=i-1&&query_mx(L[i],i-1)==a[i])L[i]=1;
    T1.tot=T2.tot=0;T1.root[0]=T2.root[0]=0;
    for(int i=1;i<=n;i++)
    {
        T1.insert(T1.root[i],1,n,L[i]),T1.merge(T1.root[i],T1.root[i-1]);
        T2.insert(T2.root[i],1,n,R[i]),T2.merge(T2.root[i],T2.root[i-1]);
    }
    int ans=0;
    while(m--)
    {
        int l=read(),r=read();
        if(type==1)l=(l+ans-1)%n+1,r=(r+ans-1)%n+1;
        if(l>r)swap(l,r);
        ans=T1.query(T1.root[l-1],T1.root[r],1,n,l+1,r)+T2.query(T2.root[l-1],T2.root[r],1,n,l,r-1);
        printf("%d\n",ans);
    }
}