[CF232D]Fence

题目大意

有一个$n$个数的数组$h$。
有$q$个询问，形如$l_0,r_0$。
每个询问查询满足一下$3$个条件的区间$[l,r]$总共有多少个？
$\bullet[l,r]\cap[l_0,r_0]=\varnothing$
$\bullet r-l=r_0-l_0$
$\bullet\forall i\in[0,r-l],h_{l_0+i}+h_{l+i}=h_{l_0}+h_l$

$1\le n,q\le10^5,1\le h_i\le10^9$

---

题目分析

将数组差分一下，即相邻两个取差变成一个新数组，我们就会发现第三个条件其实就是两个区间对应位置互为相反数。
那么我们将其取相反数后放在原数组后面，就相当于求相同子串个数。做一遍$\mathrm{SA}$即可。当然这里还有不能相交的限制，我们直接主席树，往后缀数组第$i$位主席树存入相应位置，具体细节交由读者自己想吧（不会看代码）。
时空复杂度$\mathrm O(n log_2n)$。

---

代码实现

主席树空间复杂度不是$\mathrm O(nlog_2n)$吗？为什么我开了这个大小的主席树爆掉了！！！在$\mathrm{CF}$上交了半天，还对拍了一整个晚上……最后居然是这个问题。

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cctype>
#include <cmath>

using namespace std;

int read()
{
    int x=0,f=1;
    char ch=getchar();
    while (!isdigit(ch)) f=ch=='-'?-1:f,ch=getchar();
    while (isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return x*f;
}

const int N=150000;
const int M=N<<1;
const int LGN=17;
const int LGM=18;
const int S=N*LGN;

struct chairman_tree
{
    int size[S],son[S][2];
    int tot;

    int newnode(int cp)
    {
        size[++tot]=size[cp],son[tot][0]=son[cp][0],son[tot][1]=son[cp][1];
        return tot;
    }

    void insert(int &rt,int rt0,int y,int l,int r)
    {
        rt=newnode(rt0);
        size[rt]++;
        if (l==r) return;
        int mid=l+r>>1;
        if (y<=mid) insert(son[rt][0],son[rt0][0],y,l,mid);
        else insert(son[rt][1],son[rt0][1],y,mid+1,r);
    }

    int query(int rt,int rt0,int st,int en,int l,int r)
    {
        if (!rt) return 0;
        if (st==l&&en==r) return size[rt]-size[rt0];
        int mid=l+r>>1;
        if (en<=mid) return query(son[rt][0],son[rt0][0],st,en,l,mid);
        else if (mid+1<=st) return query(son[rt][1],son[rt0][1],st,en,mid+1,r);
        else return query(son[rt][0],son[rt0][0],st,mid,l,mid)+query(son[rt][1],son[rt0][1],mid+1,en,mid+1,r);
    }
}t;

int s[M],SA[M],rank[M],height[M],Ws[M],Wv[M],x[M],y[M],root[M];
int n,q,m,lgm,ans;
int rmq[M][LGM];
int h[N];

bool cmp(int *r,int x,int y,int l){return r[x]==r[y]&&r[x+l]==r[y+l];}

bool cmp0(int x,int y){return s[x]<s[y];}

void DA()
{
    int i,l,p,mx;
    for (i=0;i<m;i++) SA[i]=i;
    sort(SA,SA+m,cmp0);
    for (x[SA[0]]=0,p=0,i=1;i<m;i++) x[SA[i]]=(p+=(s[SA[i]]!=s[SA[i-1]]));
    for (l=1;l<=m&&p!=m-1;l<<=1)
    {
        for (p=0,i=m-l;i<m;i++) y[p++]=i;
        for (i=0;i<m;i++) if (SA[i]-l>=0) y[p++]=SA[i]-l;
        for (i=0,mx=0;i<p;i++) mx=max(mx,Wv[i]=x[y[i]]);
        for (i=0;i<=mx;i++) Ws[i]=0;
        for (i=0;i<p;i++) Ws[Wv[i]]++;
        for (i=1;i<=mx;i++) Ws[i]+=Ws[i-1];
        for (i=p-1;i>=0;i--) SA[--Ws[Wv[i]]]=y[i];
        for (i=0;i<p;i++) y[i]=x[i];
        for (x[SA[0]]=0,i=1,p=0;i<m;i++) x[SA[i]]=(p+=(1^cmp(y,SA[i],SA[i-1],l)));
    }
    for (i=0;i<m;i++) rank[SA[i]]=i;
}

void getheight()
{
    height[SA[0]]=0;
    for (int i=0,j,h=0;i<m;i++,h?h--:h)
    {
        if (!rank[i]) continue;
        j=SA[rank[i]-1];
        for (;i+h<m&&j+h<m&&s[i+h]==s[j+h];) h++;
        height[rank[i]]=h;
    }
}

void pre()
{
    for (int i=0;i<m;i++) rmq[i][0]=height[i];
    lgm=trunc(log(m)/log(2));
    for (int j=1;j<=lgm;j++)
    for (int j=1;j<=lgm;j++)
        for (int i=0;i+(1<<j)-1<m;i++)
            if (rmq[i][j-1]<rmq[i+(1<<j-1)][j-1])
                rmq[i][j]=rmq[i][j-1];
            else
                rmq[i][j]=rmq[i+(1<<j-1)][j-1];
    for (int i=0;i<m;i++)
    {
        root[i]=i?root[i-1]:0;
        if (SA[i]>=n-1) t.insert(root[i],i?root[i-1]:0,SA[i]-n+1,0,n-2);
    }
}

int getrmq(int l,int r)
{
    int lgr=trunc(log(r-l+1)/log(2));
    return rmq[l][lgr]<rmq[r-(1<<lgr)+1][lgr]?rmq[l][lgr]:rmq[r-(1<<lgr)+1][lgr];
}

int solve(int L,int R)
{
    if (L>R) return n-1;
    int len=R-L+1,op,ed,l,r,mid;
    l=0,r=rank[L];
    while (l<=r)
    {
        mid=l+r>>1;
        if (getrmq(mid,rank[L])<len) op=mid,l=mid+1;
        else r=mid-1;
    }
    l=rank[L]+1,r=m-1;
    ed=rank[L];
    while (l<=r)
    {
        mid=l+r>>1;
        if (getrmq(rank[L]+1,mid)>=len) ed=mid,l=mid+1;
        else r=mid-1;
    }
    int ret=0;
    if (L-len-1>=0) ret+=t.query(root[ed],op?root[op-1]:0,0,L-len-1,0,n-2);
    if (R+2<n-1) ret+=t.query(root[ed],op?root[op-1]:0,R+2,n-2,0,n-2);
    return ret;
}

int main()
{
    freopen("fence.in","r",stdin),freopen("fence.out","w",stdout);
    n=read();
    for (int i=0;i<n;i++) h[i]=read();
    for (int i=0;i<n-1;i++) s[i]=h[i+1]-h[i];
    m=n-1<<1;
    for (int i=0;i<n-1;i++) s[i+n-1]=-s[i];
    DA(),getheight(),pre();
    q=read();
    for (int i=1,l,r;i<=q;i++)
    {
        l=read()-1,r=read()-2;
        ans=solve(l,r);
        printf("%d\n",ans);
    }
    fclose(stdin),fclose(stdout);
    return 0;
}