bzoj3926诸神眷顾的幻想乡

题意：给出一棵树，每个节点上有一个字符，问整棵树一共多少不同的串？

（保证叶子节点小于等于20）

难点1在于怎么统计到所有的子串，直接统计肯定是不行的，我们注意到叶子节点小于等于20，我们考虑将每个叶子节点作为根把树给提起来，然后定义这棵树的子串为从上到下的一个串（深度从浅到深），首先我们发现这样可以不考虑树上子串的复杂问题了（因为是有序的），其次我们发现这样之后包含了树上的所有子串（虽然我不知道为什么。）

我们统计20棵树的不同子串只需要把它们建到一个自动机上就行了。

要这样做要满足这两个条件：

1.       后缀自动机支持字符串分叉。

2.       后缀自动机支持多串匹配。

    一开始我比较怀疑这两个结论不敢乱作，不过事实证明这两个都是可以的，所以这道题剩下就简单了。

#include <iostream>#include <cstring>#include <cstdlib>#include <string>#include <cstdio>#include <algorithm>#include <cmath>#include <ctime>using namespace std;struct node{int to;int next;};node bian[500010];int tot = 1,step[8000010],son[8000010][11],pre[8000010],c,A[500010],n;int size = 0,first[500010],a,b,line[500010];long long Ans = 0;void inser(int x,int y) {bian[++size].to=y;bian[size].next=first[x];first[x]=size;}int insert(int now,int x) {int np = ++ tot;step[np] = step[now] + 1;while(son[now][x] == 0 && now != 0) son[now][x] = np,now = pre[now];if(now == 0) pre[np] = 1;else {int q = son[now][x];if(step[q] == step[now] + 1) pre[np] = q;else {int nq = ++ tot;step[nq] = step[now] + 1;for(int i = 0;i < c;i ++) son[nq][i] = son[q][i];pre[nq] = pre[q];pre[q] = pre[np] = nq;while(son[now][x] == q) son[now][x] = nq,now = pre[now];}} return np;}void dfs(int x,int Anc,int last) {int next = insert(last,A[x]);for(int u = first[x];u;u = bian[u].next)    if(bian[u].to != Anc)         dfs(bian[u].to,x,next);}int main() {scanf("%d%d",&n,&c);for(int i = 1;i <= n;i ++) scanf("%d",&A[i]);for(int i = 1;i <n;i ++) {scanf("%d%d",&a,&b);inser(a,b);inser(b,a);line[a] ++;line[b] ++;}for(int i = 1;i <= n;i ++)     if(line[i] == 1)        dfs(i,0,1);for(int i = 1;i <= tot;i ++) Ans += (long long)(step[i] - step[pre[i]]);printf("%I64d",Ans);return 0;}