记录编号	370850	评测结果	AAAAAAAAAAAAAAAAAAAA
题目名称	[ZJOI 2015]诸神眷顾的幻想乡	最终得分	100
用户昵称	AntiLeaf	是否通过	通过
代码语言	C++	运行时间	2.233 s
提交时间	2017-02-14 17:40:34	内存使用	276.90 MiB

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100010,maxm=maxn*20;
void dfs(int,int&);
void bfs();
int expand(int,int);
int trie_cnt=0,ch[maxm][10]={{0}};
int root,last[maxm],cnt=0,val[maxm<<1]={0},par[maxm<<1]={0},go[maxm<<1][10]={{0}};
vector<int>G[maxn];
long long ans=0;
int n,m,a[maxn],x,y,p[maxn],q[maxm];
int main(){
	freopen("zjoi15_substring.in","r",stdin);
	freopen("zjoi15_substring.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	for(x=1;x<=n;x++)if(G[x].size()==1u){
		fill(p,p+n+1,0);
		dfs(x,ch[0][a[x]]);
	}
	bfs();
	printf("%lld",ans);
	return 0;
}
void dfs(int x,int &rt){
	if(!rt)rt=++trie_cnt;
	for(int i=0;i<(int)G[x].size();i++)if(G[x][i]!=p[x]){
		p[G[x][i]]=x;
		dfs(G[x][i],ch[rt][a[G[x][i]]]);
	}
}
void bfs(){
	int x,head=0,tail=0;
	q[tail++]=0;
	root=last[0]=++cnt;
	while(head!=tail){
		x=q[head++];
		for(int c=0;c<m;c++)if(ch[x][c]){
			last[ch[x][c]]=expand(c,last[x]);
			q[tail++]=ch[x][c];
		}
	}
}
int expand(int c,int p){
	int np=++cnt;
	val[np]=val[p]+1;
	while(p&&!go[p][c]){
		go[p][c]=np;
		p=par[p];
	}
	if(!p)par[np]=root;
	else{
		int q=go[p][c];
		if(val[q]==val[p]+1)par[np]=q;
		else{
			int nq=++cnt;
			val[nq]=val[p]+1;
			memcpy(go[nq],go[q],sizeof(go[q]));
			par[nq]=par[q];
			par[np]=par[q]=nq;
			while(p&&go[p][c]==q){
				go[p][c]=nq;
				p=par[p];
			}
		}
	}
	ans+=val[np]-val[par[np]];
	return np;
}
/*
7 3
0 2 1 2 1 0 0
1 2
3 4
3 5
4 6
5 7
2 5
Answer:
30
*/