【bzoj3926】[Zjoi2015]诸神眷顾的幻想乡 后缀自动机

注意到叶子节点只有20个，从每个叶子节点开始dfs，在得到的trie树上建后缀自动机

比较神奇的是，我们不用考虑那么多奇怪的问题，每个节点只需要从它trie树上的父亲节点开始建后缀自动机，就是对的

讲道理，我连增量法都没理解透，更何况树上的呢，反正直接建就对了

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 100010

using namespace std;

int head[maxn],to[2*maxn],next[2*maxn],d[maxn],w[maxn];
int fa[2*20*maxn],ch[2*20*maxn][26],mx[2*20*maxn];
int n,m,num,tot=1,c;
long long ans;

void addedge(int x,int y) {num++;to[num]=y;next[num]=head[x];head[x]=num;}

int insert(int p,int x)
{
int np=++tot;
mx[np]=mx[p]+1;
while (p && !ch[p][x]) ch[p][x]=np,p=fa[p];
if (!p) fa[np]=1;
else
{
int q=ch[p][x];
if (mx[q]==mx[p]+1) fa[np]=q;
else
{
int nq=++tot;mx[nq]=mx[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q];
fa[np]=fa[q]=nq;
while (p && ch[p][x]==q) ch[p][x]=nq,p=fa[p];
}
}
return np;
}

void dfs(int x,int fa,int p)
{
int now=insert(p,w[x]);
for (int p=head[x];p;p=next[p])
if (to[p]!=fa) dfs(to[p],x,now);
}

int main()
{
scanf("%d%d",&n,&c);
for (int i=1;i<=n;i++) scanf("%d",&w[i]);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);addedge(y,x);
d[x]++;d[y]++;
}
for (int i=1;i<=n;i++) if (d[i]==1) dfs(i,0,1);
for (int i=1;i<=tot;i++) ans+=mx[i]-mx[fa[i]];
printf("%lld\n",ans);
return 0;
}