【广义后缀自动机】 [ZJOI2015]诸神眷顾的幻想乡

门
这种数据结构的话，还是封装吧。。。经常用的变量名重了会很难受的

题目

树上所有路径获得序列的本质不同串个数

题解

很难想到要用后缀自动机怎样乱搞
经过不认识的巨佬博客
了解一下广义的写法
这里用比较简单的离线，在Trie树上bfs建SAM的方法

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+10,M=2*N;
int n,m;
int s[N];
int head[N],nex[M],to[M],tot;
void build(int u,int v)
{
	tot++;nex[tot]=head[u];to[tot]=v;head[u]=tot;
}
int t[N][10],is[N],fa[N],all=1;
int add(int u,int p)
{
	if(!t[p][u])t[p][u]=++all,is[all]=u,fa[all]=p;
	return t[p][u];
}
void dfs(int u,int f,int p)
{
	int tp=add(s[u],p);
	for(int i=head[u];i;i=nex[i])
	{
		int v=to[i];if(v==f)continue;
		dfs(v,u,tp);
	}
}
//--
int size=1;
int ch[N][10],link[N],len[N];
int extend(int u,int last)
{
	int p=last,now=++size;//last=size;
	len[now]=len[p]+1;
	while(p&&!ch[p][u])ch[p][u]=now,p=link[p];
	if(!p)link[now]=1;
	else
	{
		int x=ch[p][u];
		if(len[p]+1==len[x])link[now]=x;
		else
		{
			int y=++size;
			link[y]=link[x];len[y]=len[p]+1;
			for(int i=0;i<m;i++)ch[y][i]=ch[x][i];
			while(p&&ch[p][u]==x)ch[p][u]=y,p=link[p];
			link[x]=link[now]=y;
		}
	}return now;
}
//---
int pos[N];
void bfs()
{
	int p=1;
	queue<int>q;
	for(int i=0;i<m;i++)if(t[p][i])q.push(t[p][i]);
	pos[1]=1;
	while(!q.empty())
	{
		int u=q.front();q.pop();
		pos[u]=extend(is[u],pos[fa[u]]);
		for(int i=0;i<m;i++)if(t[u][i])q.push(t[u][i]);
	}
}
int rd[N];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&s[i]);
	for(int u,v,i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		build(u,v);build(v,u);
		rd[u]++;rd[v]++;
	}
	for(int i=1;i<=n;i++)
	if(rd[i]==1)dfs(i,0,1);
	bfs();
	long long ans=0;
	for(int i=2;i<=size;i++)ans+=len[i]-len[link[i]];
	printf("%lld",ans);
}