bzoj3956 -- 单调栈 + 线段树

对于区间 [l,r] ,求出它的最大值的位置。显然不可能有点对跨越最大值。那么用单调栈预处理出每个点作为左端点、右端点的点对个数,求一个前缀和就可以了。注意相等的情况。

代码:

#include
#include
#include
#include
using namespace std;
#define N 300010
int i,j,k,n,m,s[N],Top,a[N],x,y,c[N<<2],l,r,Ans,lsum[N],rsum[N],p;
inline void Build(int x,int l,int r){
    if(l==r){
        c[x]=l;
        return;
    }
    int Mid=l+r>>1;
    Build(x<<1,l,Mid);
    Build(x<<1|1,Mid+1,r);
    c[x]=(a[c[x<<1]]>a[c[x<<1|1]]?c[x<<1]:c[x<<1|1]);
}
inline int Query(int x,int l,int r,int L,int R){
    if(l>R||rreturn 0;
    if(l>=L&&r<=R)return c[x];
    int Mid=l+r>>1;
    int f1=Query(x<<1,l,Mid,L,R),f2=Query(x<<1|1,Mid+1,r,L,R);
    return a[f1]int main(){
    scanf("%d%d%d",&n,&m,&p);
    for(i=1;i<=n;i++){
        scanf("%d",&a[i]);
        while(Top&&s[Top]if(Top)lsum[i]++;
        while(Top&&s[Top]==a[i])Top--;
        s[++Top]=a[i];
    }
    Top=0;
    for(i=n;i;i--){
        while(Top&&s[Top]if(Top)rsum[i]++;
        while(Top&&s[Top]==a[i])Top--;
        s[++Top]=a[i];
    }
    for(i=2;i<=n;i++)lsum[i]+=lsum[i-1],rsum[i]+=rsum[i-1];
    Build(1,1,n);
    while(m--){
        scanf("%d%d",&x,&y);
        if(p){
            x=(x+Ans-1)%n+1;y=(y+Ans-1)%n+1;
            if(x>y)swap(x,y);
        }
        k=Query(1,1,n,x,y);Ans=rsum[k-1]-rsum[x-1]+lsum[y]-lsum[k];
        printf("%d\n",Ans);
    }
    return 0;
}

你可能感兴趣的:(单调栈,线段树)