luoguP3346 [ZJOI2015]诸神眷顾的幻想乡

题意

学习了广义后缀自动机。

广义后缀自动机与普通后缀自动机的区别在于它是对多个串建的，于是可以处理多个串。

广义后缀自动机和普通后缀自动机的区别在于两个特判，可以见这篇题解

对于这题，因为叶子数量小于20，以每个叶子为根跑dfs，将跑到的字符串插入SAM即可，求不同子串个数是SAM的常规操作，见我的笔记。

code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2000010;
const int maxc=15;
int n,C,cnt;
int head[maxn],a[maxn],in[maxn];
struct edge{int to,nxt;}e[maxn<<1];
struct SAM
{
	int tot,last;
	int fa[maxn<<1],len[maxn<<1];
	int ch[maxn<<1][maxc];
	SAM(){last=tot=1;}
	inline void add(int c)
	{
		if(ch[last][c]&&len[ch[last][c]]==len[last]+1){last=ch[last][c];return;}
		int now=++tot;len[now]=len[last]+1;
		int p=last;
		while(p&&!ch[p][c])ch[p][c]=now,p=fa[p];
		if(!p){fa[now]=1;last=now;return;}
		int q=ch[p][c];
		bool flag=0;
		if(len[q]==len[p]+1)fa[now]=q;
		else
		{
			if(p==last)flag=1;
			int nowq=++tot;len[nowq]=len[p]+1;
			for(int i=0;i<C;i++)ch[nowq][i]=ch[q][i];
			fa[nowq]=fa[q],fa[q]=fa[now]=nowq;
			while(p&&ch[p][c]==q)ch[p][c]=nowq,p=fa[p];
			if(flag)last=nowq;
		}
		if(!flag)last=now;
	}
	inline ll query()
	{
		ll res=0;
		for(int i=2;i<=tot;i++)res+=len[i]-len[fa[i]];
		return res;
	}
}sam;
inline void add(int u,int v)
{
	e[++cnt].nxt=head[u];
	head[u]=cnt;
	e[cnt].to=v;
	in[v]++;
}
void dfs(int x,int fa)
{
	sam.add(a[x]);int last=sam.last;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(y==fa)continue;
		sam.last=last;dfs(y,x);
	}
}
int main()
{
	scanf("%d%d",&n,&C);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;scanf("%d%d",&u,&v);
		add(u,v),add(v,u);
	}
	for(int i=1;i<=n;i++)if(in[i]==1)sam.last=1,dfs(i,0);
	//cerr<<sam.tot<<endl;
	printf("%lld",sam.query());
	return 0;
}