[BZOJ3926][ZJOI2015]诸神眷顾的幻想乡（广义后缀自动机）

题目：

我是超链接

题解：

广义的后缀自动机就是对多串建立的后缀自动机。
对于这道题目来说，要求不同的子串个数，这个不就是裸的step[i]-step[fa[i]]吗，但是由于还有”顺序“问题（红黄蓝和蓝黄红），那么我们就需要对于不同的顺序建立后缀自动机，我们发现题目说叶节点不超过20个，那么我们可以从叶节点开始每次遍历树建立后缀自动机，这样得到了所有的链（即获得了所有可能的顺序），然后在后缀自动机上奔跑就可以了
trie 树的后缀自动机的状态数是线性的，不超过2|T|,|T|应该是trie的节点数。
trie树的后缀自动机的转移函数（边数）的上界为trie的节点数*字符集大小。
注意这东西既然是一棵trie树而不是很多不一样的字符串，我们就不能把last每次清1带进去玩了，而是按照树的结构一样添加东西

代码：

#include <cstdio>
#include <cstring>
#define LL long long 
using namespace std;
const int N=2000005;
int np,p,nq,q,cnt,ch[N*2][15],fa[N],tot,nxt[N],point[N],v[N],c[N],step[N],du[N];
LL ans;
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;   
}
int insert(int p,int c)
{
    np=++cnt; step[np]=step[p]+1;
    while (p && !ch[p][c]) ch[p][c]=np,p=fa[p];
    if (!p) {fa[np]=1;return np;}
    q=ch[p][c];
    if (step[q]==step[p]+1){fa[np]=q;return np;}
    nq=++cnt; step[nq]=step[p]+1;
    memcpy(ch[nq],ch[q],sizeof(ch[q]));
    fa[nq]=fa[q]; fa[q]=fa[np]=nq;
    while (ch[p][c]==q) ch[p][c]=nq,p=fa[p];
    return np;
}
void dfs(int x,int fa,int p)
{
    int t=insert(p,c[x]);
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa) dfs(v[i],x,t);
}
int main()
{
    cnt=1;int n,cc;
    scanf("%d%d",&n,&cc);
    for (int i=1;i<=n;i++) scanf("%d",&c[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addline(x,y); du[x]++; du[y]++;
    }
    for (int i=1;i<=n;i++)
      if (du[i]==1) dfs(i,0,1);

    for (int i=1;i<=cnt;i++) 
      ans+=(LL)(step[i]-step[fa[i]]);
    printf("%lld",ans);
}