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

       语文水平超级高的陈老师QAQ出题难度严格递减。。。

       实际上任意两点间的有向字符串都可以看成是任意一个叶节点为根的形成的Trie树中的一个子串。然后就可以以任意一个叶节点为跟建立后缀(chuo)自动机辣！！！（真是超级果的裸题。。然后我这么渣并不会做）。

       不需要管比如一个节点有两个完全相同的子树的情况会不会出错。。因为显然对答案没有影响。

AC代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define N 100005
using namespace std;

int n,m,tot,a[N],dgr[N],fst[N],pnt[N<<1],nxt[N<<1];
struct sam_node{
	int tot,ch[N*40][10],fa[N*40],len[N*40];
	sam_node(){ tot=1; }
	int add(int x,int p){
		int np=++tot; len[np]=len[p]+1;
		for (; !ch[p][x] && p; p=fa[p]) ch[p][x]=np;
		if (!p){ fa[np]=1; return np; }
		int q=ch[p][x];
		if (len[p]+1==len[q]) fa[np]=q; else{
			int nq=++tot; len[nq]=len[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[nq]=fa[q]; fa[q]=fa[np]=nq;
			for (; ch[p][x]==q; p=fa[p]) ch[p][x]=nq;
		}
		return np;
	}
	void qry(){
		ll ans=0; int i;
		for (i=1; i<=tot; i++) ans+=len[i]-len[fa[i]];
		printf("%lld\n",ans);
	}
}sam;
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
void add(int x,int y){
	pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot; dgr[x]++;
}
void dfs(int x,int fa,int t){
	int k=sam.add(a[x],t),p;
	for (p=fst[x]; p; p=nxt[p])
		if (pnt[p]!=fa) dfs(pnt[p],x,k);
}
int main(){
	n=read(); m=read(); int i;
	for (i=1; i<=n; i++) a[i]=read();
	for (i=1; i<n; i++){
		int x=read(),y=read();
		add(x,y); add(y,x);
	}
	for (i=1; i<=n; i++)
		if (dgr[i]==1) dfs(i,0,1);
	sam.qry();
	return 0;
}

by lych

2016.2.29