bzoj4650 [Noi2016]优秀的拆分 后缀数组

Description

如果一个字符串可以被拆分为 AABB 的形式，其中 A 和 B 是任意非空字符串，则我们称该字符串的这种拆分是优秀的。例如，对于字符串 aabaabaa，如果令 A=aab，B=a，我们就找到了这个字符串拆分成 AABB 的一种方式。一个字符串可能没有优秀的拆分，也可能存在不止一种优秀的拆分。比如我们令 A=a，B=baa，也可以用 AABB 表示出上述字符串；但是，字符串 abaabaa 就没有优秀的拆分。现在给出一个长度为 n 的字符串 S，我们需要求出，在它所有子串的所有拆分方式中，优秀拆分的总个数。这里的子串是指字符串中连续的一段。以下事项需要注意：出现在不同位置的相同子串，我们认为是不同的子串，它们的优秀拆分均会被记入答案。在一个拆分中，允许出现 A=B。例如 cccc 存在拆分 A=B=c。字符串本身也是它的一个子串。
T<=10,n<=30000

Solution

---

挺神的题，想不到这种奇迹银翘

首先容易想到枚举分割点，设f[i]表示以第i位为结尾的AA串的数量，g[i]表示以第i位为开头的BB串的数量，那么答案就是$\sum _{i=1}^{n-1}f[i]\ast g[i+1]$

考虑怎么求这个东西。我们枚举AA串长度的一半len，每隔len个点就标记一下，这样得到n/len个点
显然长度恰好为2len的AA串一定经过两个标记点，那么我们就枚举相邻的点求后缀的lcp和前缀的lcs，拼在一起后可以发现合法的起点和终点都是一整段，我们维护差分数组然后做一次前缀和就可以得到f和g了

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))

typedef long long LL;
const int N=30005;

int n;

struct SA {
	int sa[N],rank[N],h[N],r1[N],s2[N],ct[N];
	int mn[15][N];
	char str[N];
	void pre() {
		fill(ct,0); fill(rank,0);
		fill(sa,0); fill(h,0);
		fill(r1,0); fill(s2,0);
		fill(mn,0);
		rep(i,1,n) ct[rank[i]=str[i]-'a'+1]++;
		rep(i,1,26) ct[i]+=ct[i-1];
		drp(i,n,1) sa[ct[rank[i]]--]=i;
		for (int j=1,k=0,mx=26;k<n;j<<=1,mx=k) {
			int p=0;
			rep(i,n-j+1,n) s2[++p]=i;
			rep(i,1,n) if (sa[i]>j) s2[++p]=sa[i]-j;
			fill(ct,0);
			rep(i,1,n) ct[rank[s2[i]]]++;
			rep(i,1,mx) ct[i]+=ct[i-1];
			drp(i,n,1) sa[ct[rank[s2[i]]]--]=s2[i];
			r1[sa[1]]=k=1;
			rep(i,2,n) {
				if (rank[sa[i-1]]==rank[sa[i]]&&rank[sa[i-1]+j]==rank[sa[i]+j]) {
					r1[sa[i]]=k;
				} else r1[sa[i]]=++k;
			}
			rep(i,1,n) rank[i]=r1[i];
		}
		// rep(i,1,n) printf("%d ", rank[i]); puts("");
		h[1]=0;
		for (int i=1,j=0;i<=n;++i) {
			if (rank[i]==1) continue;
			j=std:: max(h[rank[i-1]]-1,0);
			while (str[i+j]==str[sa[rank[i]-1]+j]) j++;
			h[rank[i]]=j;
		}
		rep(i,1,n) mn[0][i]=h[i];
		rep(j,1,14) {
			int len=1<<j-1;
			rep(i,1,n+1-len) mn[j][i]=std:: min(mn[j-1][i],mn[j-1][i+len]);
		}
	}
	int lcp(int l,int r) {
		l=rank[l],r=rank[r];
		if (r<l) std:: swap(l,r); l++;
		int lg=(int)log2(r-l+1);
		return std:: min(mn[lg][l],mn[lg][r-(1<<lg)+1]);
	}
} S,T;

char str[N];

LL f[N],g[N];

int main(void) {
	freopen("data.in","r",stdin);
	freopen("myp.out","w",stdout);
	int Q; scanf("%d",&Q);
	for (;Q--;) {
		scanf("%s",str+1); n=strlen(str+1);
		fill(S.str,0),fill(T.str,0);
		f[0]=g[0]=0;
		rep(i,1,n) {
			S.str[i]=str[i];
			T.str[i]=str[n-i+1];
			f[i]=g[i]=0;
		}
		S.pre(); T.pre();
		rep(len,1,n) {
			for (int i=1;i+len<=n;i+=len) {
				int j=i+len;
				int w1=S.lcp(i,j);
				int w2=T.lcp(n-i+1,n-j+1);
				w1=std:: min(w1,len);
				w2=std:: min(w2,len);
				if (w1+w2-1<len) continue;
				f[i-w2+1]++,f[i+w1-len+1]--;
				g[j-w2+len]++,g[i+w1+len]--;
			}
		}
		// rep(i,1,n) printf("%lld %lld\n", f[i],g[i]);
		for (int i=1;i<=n;++i) printf("%lld %lld\n", f[i],g[i]);
		rep(i,1,n) f[i]+=f[i-1],g[i]+=g[i-1];
		LL ans=0;
		rep(i,1,n-1) ans+=f[i+1]*g[i];
		printf("%lld\n", ans);
	}
	return 0;
}