题目大意
有一个
n
个数的数组
有
q
个询问,形如
每个询问查询满足一下
3
个条件的区间
∙[l,r]∩[l0,r0]=∅
∙r−l=r0−l0
∙∀i∈[0,r−l],hl0+i+hl+i=hl0+hl
1≤n,q≤105,1≤hi≤109
题目分析
将数组差分一下,即相邻两个取差变成一个新数组,我们就会发现第三个条件其实就是两个区间对应位置互为相反数。
那么我们将其取相反数后放在原数组后面,就相当于求相同子串个数。做一遍
SA
即可。当然这里还有不能相交的限制,我们直接主席树,往后缀数组第
i
位主席树存入相应位置,具体细节交由读者自己想吧(不会看代码)。
时空复杂度
代码实现
主席树空间复杂度不是 O(nlog2n) 吗?为什么我开了这个大小的主席树爆掉了!!!在 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;
}