BZOJ 2119: 股市的预测 【枚举长度找关键点+后缀数组求LCP】

题意：

给一个长度为N的A数组，一个时间间隔M，要你把A数组的每个元素与前一个作差得到一个长度为N-1的B数组。在B数组中找形如aba的子串的个数，其中b的长度为M。N,M<=50000

题目分析：

在b的长度一定的情况下，怎么找形如aba的串？
枚举a的长度，设为len，把序列分成这样的几段：

那么a串一定包含上面画的关键点中的某一个。
只需要枚举这些点，假设枚举到点i，那么只需要看i与i+M+len对应的串往后和往前最长能够匹配多少（求正串及反串的LCP），就可以知道包含点i的长度为len的a串的个数了。具体看下面代码中的solve函数。

时间复杂度如下：

Code：

#include<cstdio>
#include<cctype>
#include<algorithm>
#define maxn 100005
using namespace std;
char cb[1<<15],*cs,*ct;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<15,stdin),cs==ct)?0:*cs++)
inline void read(int &a){
	char c;while(!isdigit(c=getc()));
	for(a=c-'0';isdigit(c=getc());a=a*10+c-'0');
}
int ary[4][maxn],b[maxn],height[maxn],st[maxn][18],lg[maxn];
int *sa=ary[0],*rk=ary[1],*nsa=ary[2],*nrk=ary[3];
void cal_sa(int n,int m,int *a){
	for(int i=1;i<=n;i++) b[a[i]]++;
	for(int i=1;i<=m;i++) b[i]+=b[i-1];
	for(int i=1;i<=n;i++) sa[b[a[i]]--]=i;
	for(int i=1;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(a[sa[i-1]]!=a[sa[i]]);
	for(int k=1;k<n&&rk[sa[n]]<n;k<<=1){
		for(int i=1;i<=n;i++) b[rk[sa[i]]]=i;
		for(int i=n;i>=1;i--) if(sa[i]>k) nsa[b[rk[sa[i]-k]]--]=sa[i]-k;
		for(int i=n-k+1;i<=n;i++) nsa[b[rk[i]]--]=i;
		for(int i=1;i<=n;i++) nrk[nsa[i]]=nrk[nsa[i-1]]+(rk[nsa[i-1]]!=rk[nsa[i]]||rk[nsa[i-1]+k]!=rk[nsa[i]+k]);
		swap(sa,nsa),swap(rk,nrk);
	}
}
void cal_height(int n,int *a){
	for(int i=1,k=0;i<=n;i++) if(rk[i]!=1){
		if(k) k--;
		for(int j=sa[rk[i]-1];a[j+k]==a[i+k];k++);
		height[rk[i]]=k;
	}
}
void STpre(int n){
	lg[0]=-1;
	for(int i=1;i<=n;i++) st[i][0]=height[i],lg[i]=lg[i>>1]+1;
	for(int j=1;(1<<j)<=n;j++) for(int i=1;i+(1<<(j-1))<=n;i++) st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
inline int LCP(int l,int r){
	l=rk[l],r=rk[r];if(l>r) swap(l,r);l++;
	int k=lg[r-l+1];
	return min(st[l][k],st[r-(1<<k)+1][k]);
}
int n,m,ans,a[maxn],tmp[maxn];
void solve(){
	for(int l=1;l+l+m<=n;l++)
		for(int i=1,j;i+l+m<=n;i+=l) if(a[i]==a[j=i+l+m]){
			int x=min(l,LCP(i,j)),y=min(l,LCP(n+n+2-i,n+n+2-j));
			if(x+y>l) ans+=x+y-l;
		}
}
int main()
{
	read(n),read(m);n--;
	for(int i=1;i<=n+1;i++) read(a[i]);
	for(int i=1;i<=n;i++) a[i]=a[i+1]-a[i],tmp[i]=a[i];
	sort(tmp+1,tmp+1+n);
	int len=unique(tmp+1,tmp+1+n)-tmp-1;
	for(int i=1;i<=n;i++) a[i]=lower_bound(tmp+1,tmp+1+len,a[i])-tmp;
	a[n+1]=len+1;for(int i=1;i<=n;i++) a[n+i+1]=a[n-i+1];
	cal_sa(n+n+1,len+1,a), cal_height(n+n+1,a), STpre(n+n+1);
	solve();
	printf("%d",ans);
}