bzoj 3926: 诸神眷顾的幻想乡 广义后缀自动机

题目:

Description

幽香是全幻想乡里最受人欢迎的萌妹子，这天，是幽香的2600岁生日，无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日。 粉丝们非常热情，自发组织表演了一系列节目给幽香看。幽香当然也非常高兴啦。 这时幽香发现了一件非常有趣的事情，太阳花田有n块空地。在过去，幽香为了方便，在这n块空地之间修建了n-1条边将它们连通起来。也就是说，这n块空地形成了一个树的结构。 有n个粉丝们来到了太阳花田上。为了表达对幽香生日的祝贺，他们选择了c中颜色的衣服，每种颜色恰好可以用一个0到c-1之间的整数来表示。并且每个人都站在一个空地上，每个空地上也只有一个人。这样整个太阳花田就花花绿绿了。幽香看到了，感觉也非常开心。 粉丝们策划的一个节目是这样的，选中两个粉丝A和B（A和B可以相同），然后A所在的空地到B所在的空地的路径上的粉丝依次跳起来（包括端点），幽香就能看到一个长度为A到B之间路径上的所有粉丝的数目（包括A和B）的颜色序列。一开始大家打算让人一两个粉丝（注意：A,B和B,A是不同的，他们形成的序列刚好相反，比如红绿蓝和蓝绿红）都来一次，但是有人指出这样可能会出现一些一模一样的颜色序列，会导致审美疲劳。 于是他们想要问题，在这个树上，一共有多少可能的不同的颜色序列（子串）幽香可以看到呢？ 太阳花田的结构比较特殊，只与一个空地相邻的空地数量不超过20个。

Input

第一行两个正整数n,c。表示空地数量和颜色数量。

第二行有n个0到c-1之间，由空格隔开的整数，依次表示第i块空地上的粉丝的衣服颜色。（这里我们按照节点标号从小到大的顺序依次给出每块空地上粉丝的衣服颜色）。

接下来n-1行，每行两个正整数u,v，表示有一条连接空地u和空地v的边。

Output

一行，输出一个整数，表示答案。

题解:

我们发现很关键的一点 : 只与一个空地相邻的空地数量不超过20个

也就是说叶子节点最多有20个

所以我们想到了将所有的串提取出来的方法:

分别以每个叶子为根,用所有的树共同建立广义后缀自动机

然后直接统计所有的\(len_x - len_{fa_x}\)即可

其实这是论文题:

见:

2015 年国家集训队论文

《后缀自动机在字典树上的拓展》 刘研绎

《后缀自动机及其应用》 张天扬

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
inline void read(int &x){
	x=0;char ch;bool flag = false;
	while(ch=getchar(),ch<'!');if(ch == '-') ch=getchar(),flag = true;
	while(x=10*x+ch-'0',ch=getchar(),ch>'!');if(flag) x=-x;
}
const int maxn = 100010;
struct Edge{
	int to,next;
}G[maxn<<1];
int head[maxn],cnt,maxc;
void add(int u,int v){
	G[++cnt].to = v;
	G[cnt].next = head[u];
	head[u] = cnt;
}
int w[maxn],deg[maxn];
struct Node{
	int nx[10];
	int len,fa;
}T[maxn*20*2];
int nodecnt;
inline int insert(int last,int c){
	int cur = ++nodecnt,p;
	T[cur].len = T[last].len + 1;
	for(p = last;p != -1 && !T[p].nx[c];p = T[p].fa) T[p].nx[c] = cur;
	if(p == -1) T[p].fa = 0;
	else{
		int q = T[p].nx[c];
		if(T[q].len == T[p].len + 1) T[cur].fa =q;
		else{
			int co = ++ nodecnt;
			T[co] = T[q];T[co].len = T[p].len + 1;
			for(;p != -1 && T[p].nx[c] == q;p = T[p].fa) T[p].nx[c] = co;
			T[cur].fa = T[q].fa = co;
		}
	}return cur;
}
#define v G[i].to
void dfs(int u,int fa,int last){
	last = insert(last,w[u]);
	for(int i = head[u];i;i=G[i].next){
		if(v == fa) continue;
		dfs(v,u,last);
	}
}
#undef v
inline void init(){
	nodecnt = 0;
	T[0].fa = -1;
}
int main(){
	init();int n;read(n);read(maxc);
	for(int i=1;i<=n;++i) read(w[i]);
	for(int i=1,u,v;i<n;++i){
		read(u);read(v);
		add(u,v);add(v,u);
		++deg[u];++deg[v];
	}
	for(int i=1;i<=n;++i){
		if(deg[i] == 1) dfs(i,0,0);
	}ll ans = 0;
	for(int i=1;i<=nodecnt;++i){
		ans += (T[i].len - T[T[i].fa].len);
	}printf("%lld\n",ans);
	getchar();getchar();
	return 0;
}