【BZOJ】4650: [Noi2016]优秀的拆分 -后缀数组

传送门：bzoj4650

---

题解

考虑枚举对称点，设$q{z}_i,h{z}_i$分别为以$i$结尾/开头的形式为$AA$的字符串的个数，则$ans=\sum _{i=1}^{n-1}q{z}_i\times h{z}_{i+1}$

考虑一个长为$l$的$AA$串必然经过至少2个原串中$i=k\times \frac{l}{2}$的位置，所以可以$O(n\ln n)$枚举枚举$AA$串中$A$的长度$l$以及$i=k\times l,j=i+l,j\le n$处对答案的贡献。

设$x=lcs(pre(i-1),pre(j-1)),y=lcp(suf(i),suf(j))$，若$x+y\ge l$，则总共存在$x+y-l+1$个覆盖这两个端点长度为$2l$的$AA$串，等价于区间$f,g$的+1操作，差分左端点+1，右端点后-1，最后求前缀和即可。

快速求前后缀$lcp,lcs$正反做2遍后缀数组后$st$表记录即可。

一些需要注意的细节：

1.计算$x=lcs(pre(i-1),pre(j-1)),y=lcp(suf(i),suf(j))$有可能$x\ge l$或者$y>l$，导致枚举到的某些区间实际上并不完全覆盖这两个点，不能当前点对的计入贡献。所以$x,y$要分别与$l-1,l$取$min$。
2.多组询问时记得清空$SA$中两个桶数组。(不清空判断y[sa[i]+k]!=y[sa[i-1]+k]时会爆掉)。

---

代码

#include<bits/stdc++.h>
#define mem(f,x) memset((f),(x),sizeof((f)))
using namespace std;
typedef long long ll;
const int N=6e4+10;

int tk,n,m,bin[30],lg[N];
int qz[N],hz[N];char s[N];ll ans;

struct SA{
	int m,sa[N],t1[N],t2[N],c[N],rk[N];
	int ht[17][N];
	inline void itia(){mem(ht,0x7f);mem(t1,0);mem(t2,0);}
	
    inline void build()
    {
    	int i,k,p,*x=t1,*y=t2;m=26;
    	for(i=1;i<=m;++i) c[i]=0;
    	for(i=1;i<=n;++i) c[(x[i]=(int)s[i])]++;
    	for(i=1;i<=m;++i) c[i]+=c[i-1];
		for(i=n;i;--i) sa[c[x[i]]--]=i;
		for(k=1;k<n;k<<=1){
	        
	        //the second key
			for(p=0,i=n-k+1;i<=n;++i) y[++p]=i;
	        for(i=1;i<=n;++i) if(sa[i]>k) y[++p]=sa[i]-k;
	        
	        //the first key
			for(i=1;i<=m;++i) c[i]=0;
			for(i=1;i<=n;++i) c[x[y[i]]]++;
			for(i=1;i<=m;++i) c[i]+=c[i-1];
			for(i=n;i;--i) sa[c[x[y[i]]]--]=y[i];
			
			p=1;swap(x,y);x[sa[1]]=1;
			for(i=2;i<=n;++i){
				p+=((y[sa[i]]!=y[sa[i-1]])||(y[sa[i]+k]!=y[sa[i-1]+k]))?1:0;
				x[sa[i]]=p;
			}
			if(p>=n) break;
			m=p;
		}
    }
    inline void mk()
    {
    	int i,j,k=0;
    	for(i=1;i<=n;++i) rk[sa[i]]=i;
    	for(i=1;i<=n;++i){
    		if(rk[i]==1) {k=0;continue;}
			if(k) k--;j=sa[rk[i]-1];
    		for(;j+k<=n && i+k<=n && s[j+k]==s[i+k];++k);
    		ht[0][rk[i]]=k;
    	}
    	for(k=1;bin[k]<=n;++k){
    		for(i=1;i+bin[k]-1<=n;++i)
    			ht[k][i]=min(ht[k-1][i],ht[k-1][i+bin[k-1]]);
    	}
    }
    inline int ask(int x,int y)
    {
    	x=rk[x];y=rk[y];if(x>y) swap(x,y);
    	int cg=lg[y-x];
    	return min(ht[cg][x+1],ht[cg][y-bin[cg]+1]);
    }
}A,B;

inline void sol()
{
	int i,j,x,y,z,l;
	A.itia();B.itia();mem(qz,0);mem(hz,0);
	scanf("%s",s+1);n=strlen(s+1);
	for(i=1;i<=n;++i) s[i]=s[i]-'a'+1;
	A.build();A.mk();
	reverse(s+1,s+n+1);
	B.build();B.mk();
	for(l=1;l+l<=n;++l)
		for(i=l,j=l+l;j<=n;i+=l,j+=l){
			x=min(B.ask(n+2-j,n+2-i),l-1);y=min(A.ask(i,j),l);
			if(x+y<l) continue;z=x+y-l+1;
			qz[j+y-z]++;qz[j+y]--;
			hz[i-x]++;hz[i-x+z]--;
		}
	for(i=1;i<=n;++i) qz[i]+=qz[i-1],hz[i]+=hz[i-1];
	ans=0LL;
	for(i=1;i<n;++i) ans+=(ll)qz[i]*hz[i+1];
	printf("%lld\n",ans);
}

int main(){
	int i;bin[0]=1;
	for(i=1;i<25;++i) bin[i]=bin[i-1]<<1;
	for(i=2;i<N;++i) lg[i]=lg[i>>1]+1;
	for(scanf("%d",&tk);tk;--tk) sol();
	return 0;
}