CodeForces 232D

后缀数组+rmq+树状数组

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
const int N=2e5+100;
int na[N],nb[N],ns[N],nv[N];
int sa[N],he[N],rk[N];
int cmp(int *r,int a,int b,int l){
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void _sort(int x[N],int y[N],int n,int m){
    for(int i=0;i<m;i++)ns[i]=0;
    for(int i=0;i<n;i++)ns[x[y[i]]]++;
    for(int i=1;i<m;i++)ns[i]+=ns[i-1];
    for(int i=n-1;i>=0;i--)
        sa[--ns[x[y[i]]]]=y[i];
}
void da(int *r,int n,int m){
    int *x=na,*y=nb;
    for(int i=0;i<n;i++)x[i]=r[i];
    for(int i=0;i<n;i++)y[i]=i;
    _sort(x,y,n,m);
    for(int j=1,p=1,i; p<n; j*=2, m=p){
        for(p=0, i=n-j; i<n; i++)y[p++]=i;
        for(i=0; i<n; i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        _sort(x,y,n,m);
        swap(x,y);p=1;x[sa[0]]=0;
        for(i=1;i<n;i++){
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        }
    }
}
void gethe(int *r,int n){
    for(int i=1;i<=n;i++)rk[sa[i]]=i;
    for(int i=0, k=0, j=0; i<n; he[rk[i++]]=k){
        for(k?k--:0, j=sa[rk[i]-1]; r[i+k]==r[j+k]; k++);
    }
}
void pri(int aa[N],int n){
    for(int i=0;i<n;i++){
        printf("%d ",aa[i]);
    }printf("\n");
}
void solve(int *r,int n){
    r[n]=0;
    da(r,n+1,n+1);
    gethe(r,n);
//    pri(rk,n+1);
//    pri(sa,n+1);
//    pri(he,n+1);
}
int lg[N],mi[N];
void init(){
    int n=2e5;
    mi[0]=1;
    for(int i=1;i<=20;i++){
        mi[i]=mi[i-1]*2;
    }
    lg[0]=-1;
    int cc=1;
    for(int i=1;i<=n;i++){
        if(i==cc){
            lg[i]=lg[i-1]+1,cc*=2;
        }
        else lg[i]=lg[i-1];
    }
}

int dp[N][22];
void _bei(int n){
    for(int i=0;i<n;i++)dp[i][0]=he[i];
    for(int i=1;i<=lg[n];i++){
        for(int j=0;j+mi[i]-1<n;j++){
            dp[j][i]=min(dp[j][i-1],dp[j+mi[i-1]][i-1]);
        }
    }
}

int query(int l,int r){
    int k=lg[r-l+1];
    return min(dp[l][k],dp[r-mi[k]+1][k]);
}

int aa[N],bb[N];
void build(int n){
    for(int i=0;i<n;i++){
        scanf("%d",&aa[i]);
    }
    for(int i=1;i<n;i++){
        bb[i]=aa[i]-aa[i-1];
    }
    bb[0]=aa[0];
    for(int i=n;i<2*n-1;i++){
        bb[i]=aa[i-n]-aa[i+1-n];
    }
    bb[2*n-1]=aa[n-1];
    for(int i=0;i<2*n;i++){
        aa[i]=bb[i];
    }
    sort(aa,aa+2*n);
    //pri(bb,2*n+1);
    int top=unique(aa,aa+2*n)-aa;
    for(int i=0;i<2*n;i++){
        bb[i]=lower_bound(aa,aa+2*n,bb[i])-aa+1;
    }
    //pri(bb,2*n+1);
    solve(bb,2*n);
    _bei(2*n);
}

int ans[N];
vector<int> vv[N];
#define pb push_back
int cc[N];
void Ins(int x,int n){
    while(x<=n){
        cc[x]++;x+=x&-x;
    }
}
int query(int x){
    int sum=0;
    while(x){
        sum+=cc[x];x-=x&-x;
    }
    return sum;
}

int ll[N],rr[N];
void geta(int n){
    memset(cc,0,sizeof(cc));
    for(int i=0;i<n;i++){
        Ins(rk[i+n],2*n);
        for(int j=0;j<vv[i].size();j++){
            int op=vv[i][j]>0?1:-1;
            int id=abs(vv[i][j]);
            int x=query(rr[id])-query(ll[id]-1);
            ans[id]+=op*x;
        }
    }
}

int main(){
    #ifdef DouBi
    freopen("in.cpp","r",stdin);
    #endif // DouBi
    int n;
    init();
    while(scanf("%d",&n)!=EOF){

        build(n);

        for(int i=0;i<2*n;i++){
            vv[i].clear();
        }
        int q;scanf("%d",&q);
        for(int i=1;i<=q;i++){
            int l,r;scanf("%d%d",&l,&r);l--;r--;
            if(l==r){
                ans[i]=n-1;continue;
            }
            int len=r-l+1;
            if(l-len>=0){
                vv[l-len].pb(i);
            }
            if(r+1<=n-len){
                vv[r].pb(-i);
                vv[n-len].pb(i);
            }

            l++;
            int x1=rk[l];
            for(int j=19;j>=0;j--){
                if(x1+mi[j]>=2*n)continue;
                if(query(x1+1,x1+mi[j])<r-l+1)continue;
                x1=x1+mi[j];
            }
            int y1=rk[l];
            for(int j=19;j>=0;j--){
                if(y1-mi[j]<0)continue;
                if(query(y1-mi[j]+1,y1)<r-l+1)continue;
                y1=y1-mi[j];
            }
            ll[i]=y1,rr[i]=x1;
        }

        geta(n);
        for(int i=1;i<=q;i++){
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}