BZOJ　3238: [Ahoi2013]差异（后缀自动机构建后缀树）

题目
这个式子一看好眼熟。
就是求后缀树上所有后缀所在点的距离之和。
后缀树有一个（自认为）很恶心优秀的$O(n)$O(n)
后缀自动机求出来的后缀链接其实是前缀树。。。。。。
把字符串反过来就能建树了。
$O(n*26)$O(n∗26)//$26$26是复制前向边的复杂度。

AC Code:

#include<bits/stdc++.h>
#define maxn 1000005
#define maxc 26
#define LL long long
using namespace std;

int tr[maxn][maxc],fail[maxn],len[maxn];
int tot;
void ins(int p,int c){
	int cur=++tot,q;
	len[cur]=len[p]+1;
	for(;p!=-1&&!tr[p][c];p=fail[p]) tr[p][c]=cur;
	if(p==-1) fail[cur]=0;
	else if(len[q=tr[p][c]]==len[p]+1) fail[cur]=q;
	else{ int cln=++tot;
		memcpy(tr[cln],tr[q],sizeof tr[q]),fail[cln]=fail[q],len[cln]=len[p]+1;
		for(;p!=-1&&tr[p][c]==q;p=fail[p]) tr[p][c]=cln;
		fail[cur]=fail[q]=cln;
	}
}
char s[maxn];
int n,cnt[maxn],c[maxn],sa[maxn];
int main(){
	scanf("%s",s);n=strlen(s);
	fail[0]=-1;
	for(int i=0,p=0;i<n;i++){
		ins(p,s[n-i-1]-'a');
		p=tr[p][s[n-i-1]-'a'];
		cnt[p]++;
	}
	for(int i=1;i<=tot;i++) c[len[i]]++;
	for(int i=1;i<=n;i++) c[i]+=c[i-1];
	for(int i=1;i<=tot;i++) sa[c[len[i]]--]=i;
	LL ans=0;
	for(int i=tot;i>=1;i--){
		int u = sa[i];
		ans += 1ll * cnt[u] * (n-cnt[u]) * (len[u]-len[fail[u]]);
		cnt[fail[u]] += cnt[u];
	}
	printf("%lld\n",ans);
}

标签：后缀,3238,tr,Ahoi2013,len,int,maxn,fail
来源： https://blog.csdn.net/qq_35950004/article/details/88876422