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

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #include<cstring>
 5 #include<cmath>
 6 #include<queue>
 7 using namespace std;
 8 #define LL long long
 9 const int maxn=100100;
10 int n,c;
11 int val[maxn]={};
12 struct nod{
13     int y,next;
14 }e[maxn*2];int head[maxn]={},tot=0;
15 int a[maxn]={};
16 struct node{
17     int sig[15],f,len;
18 }t[maxn*40];int tly=1;
19 inline void init(int x,int y){ e[++tot].y=y; e[tot].next=head[x]; head[x]=tot; }
20 int addit(int k,int z){
21     int x=++tly; t[x].len=t[k].len+1;
22     for(;k&&!t[k].sig[z];k=t[k].f)
23         t[k].sig[z]=x;
24     if(k==0)t[x].f=1;
25     else{
26         int y=t[k].sig[z];
27         if(t[y].len==t[k].len+1)t[x].f=y;
28         else{
29             int p=++tly; t[p]=t[y]; t[p].len=t[k].len+1;
30             t[x].f=t[y].f=p;
31             for(;k&&t[k].sig[z]==y;k=t[k].f)
32                 t[k].sig[z]=p;
33         }
34     }
35     return x;
36 }
37 void dfs(int x,int fa,int k){
38     int nt=addit(k,val[x]);
39     for(int i=head[x];i;i=e[i].next)
40         if(fa!=e[i].y)dfs(e[i].y,x,nt);
41 }
42 int main(){
43     memset(t,0,sizeof(t));
44     int x,y;
45     scanf("%d%d",&n,&c);
46     for(int i=1;i<=n;++i)scanf("%d",&val[i]);
47     for(int i=1;i<n;++i){
48         scanf("%d%d",&x,&y);
49         init(x,y);init(y,x); a[x]++;a[y]++;
50     }
51     for(int i=1;i<=n;++i)if(a[i]==1)dfs(i,0,1);
52     LL ans=0;
53     for(int i=1;i<=tly;++i)ans+=(LL)(t[i].len-t[t[i].f].len);
54     printf("%lld\n",ans);
55     return 0;
56 }