BZOJ 3926: [Zjoi2015]诸神眷顾的幻想乡【广义后缀自动机】

题目传送门

题目分析：

这道题就是求多个串的不同子串的个数。
广义后缀自动机，就是可以表示所有串的所有后缀的自动机。
其实就是建完一个串之后，把另一个串接在初始节点后面建就好了。

看完这句话的第一反应？
点分治!
不是度数不超过20，是叶子数不超过20。。。
所以从每个叶子跑一个SAM，把它们合并起来变成一个大SAM，统计在这个大SAM上不同的子串个数。
怎么统计？
蒟蒻知道两种方法 ：

• “每个点的len值减去它后缀链接指向的点的len值” 之和
• 每个点表示一个串，cnt为1（根节点为0），每个点的sum为自己的cnt加上后继的sum，最后答案为根节点的sum

Code：

#include<cstdio>
#include<cstring>
#include<cctype>
#define maxn 4000005
#define N 100005
int maxc;
inline void read(int &a){
	char c;while(!isdigit(c=getchar()));
	for(a=c-'0';isdigit(c=getchar());a=a*10+c-'0');
}
int n,a[N],in[N];
int fir[N],nxt[N<<1],to[N<<1],tot;
inline void line(int x,int y){nxt[++tot]=fir[x];fir[x]=tot;to[tot]=y;}
int fail[maxn]={-1},ch[maxn][10],len[maxn],cnt[maxn],sz,last;
void sa_extend(int c){
	int p=last,cur=++sz,q;
	len[last=cur]=len[p]+1,cnt[cur]=1;
	for(;p!=-1&&!ch[p][c];p=fail[p]) ch[p][c]=cur;
	if(p==-1) fail[cur]=0;
	else if(len[p]+1==len[q=ch[p][c]]) fail[cur]=q;
	else{
		int clone=++sz; len[clone]=len[p]+1,fail[clone]=fail[q];
		fail[q]=fail[cur]=clone;
		memcpy(ch[clone],ch[q],sizeof ch[q]);
		for(;p!=-1&&ch[p][c]==q;p=fail[p]) ch[p][c]=clone;
	}
}
void dfs(int u,int pre){
	int tmp=last;
	for(int i=fir[u];i;i=nxt[i]) if(to[i]!=pre)
		last=tmp,sa_extend(a[to[i]]),dfs(to[i],u);
}
int main()
{
	read(n),read(maxc);
	for(int i=1;i<=n;i++) read(a[i]);
	for(int i=1,x,y;i<n;i++) read(x),read(y),line(x,y),line(y,x),in[x]++,in[y]++;
	for(int i=1;i<=n;i++) if(in[i]==1) last=0,sa_extend(a[i]),dfs(i,0);
	long long ans=0;
	for(int i=1;i<=sz;i++) ans+=len[i]-len[fail[i]];
	printf("%lld",ans);
}

PS：在BZOJ上空间开大了会CE，上面代码的maxn改成2000000才能过。。