记录编号	370839	评测结果	AAAAAAAAAA
题目名称	[河南省队2016]九头蛇和时间	最终得分	100
用户昵称	AntiLeaf	是否通过	通过
代码语言	C++	运行时间	0.376 s
提交时间	2017-02-14 17:08:50	内存使用	8.70 MiB

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
const int maxn=100010;
void bfs();
int expand(int,int);
int root,last[maxn],cnt=0,val[maxn<<1]={0},par[maxn<<1]={0};
map<int,int>go[maxn<<1];
vector<int>G[maxn];
long long ans=0;
int n,p[maxn]={0},q[maxn],x,y;
int main(){
	freopen("af_time.in","r",stdin);
	freopen("af_time.out","w",stdout);
	root=last[0]=++cnt;
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	bfs();
	printf("%lld",ans);
	return 0;
}
void bfs(){
	int x,head=0,tail=0;
	q[tail++]=1;
	while(head!=tail){
		x=q[head++];
		last[x]=expand(G[x].size(),last[p[x]]);
		for(int i=0;i<(int)G[x].size();i++)if(G[x][i]!=p[x]){
			p[G[x][i]]=x;
			q[tail++]=G[x][i];
		}
	}
}
int expand(int c,int p){
	int np=++cnt;
	val[np]=val[p]+1;
	while(p&&!go[p].count(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;
			go[nq]=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;
}