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

题目大意：
给定一棵n（1<=n<=100000）个点的树，每个点上有一种颜色，颜色数不超过10。
问这个树上有多上个本质不同的链。
只与一条边相连的点不超过20个。

题目分析：（广义后缀自动机）
广义后缀自动机啥的听起来就很高端，但是其实就是把几个串的的root捏在一起做后缀自动机……
因为只与一条边相连的点不超过20个，所以建20个广义后缀自动机，在插入点的到时候顺便维护一下答案。也就是20遍dfs，时间复杂度O（n）。

注意事项： 答案输出用long long（我这个不长记性的大二货QAQ……）
代码如下：

#include<cstdio>
#include<cstring>
#define N 120000
using namespace std;
int n,m;
long long ans;
int fir[N],nes[N<<1],v[N<<1],top=1,q[N];
int in_d[N];
void edge(int x,int y)
{
    top++;
    v[top]=y;
    nes[top]=fir[x];
    fir[x]=top;
    in_d[y]++;
    return;
}
#define edge(x,y) edge(x,y),edge(y,x)

struct SAM{
    SAM *son[10],*fa;
    int max_len;
    SAM(int _=0):fa(0x0),max_len(_){memset(son,0,sizeof(son));}
}*root=new SAM,*last[N];
SAM* extend(int x,SAM *last)
{
    SAM *p=last;
    SAM *np=new SAM(p->max_len+1);
    while(p && !p->son[x]) p->son[x]=np,p=p->fa;
    if(!p) np->fa=root;
    else
    {
        SAM *q=p->son[x];
        if(p->max_len+1==q->max_len) np->fa=q;
        else
        {
            SAM *nq=new SAM(p->max_len+1);
            ans-=q->max_len-q->fa->max_len;
            nq->fa=q->fa;
            memcpy(nq->son,q->son,sizeof(nq->son));
            q->fa=nq; np->fa=nq;
            for(;p && p->son[x]==q;p=p->fa) p->son[x]=nq;
            ans+=q->max_len-q->fa->max_len;
            ans+=nq->max_len-nq->fa->max_len;
        }
    }
    ans+=np->max_len-np->fa->max_len;
    return np;
}

void dfs(int c,int father)
{
    last[c]=extend(q[c],last[father]);
    for(int t=fir[c];t;t=nes[t])
        if(v[t]!=father) dfs(v[t],c);
    return;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&q[i]);
    for(int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        edge(x,y);
    }
    last[0]=root;
    for(int i=1;i<=n;i++)
        if(in_d[i]==1) dfs(i,0);
    printf("%lld\n",ans);
    return 0;
}