bzoj3926 [Zjoi2015]诸神眷顾的幻想乡

http://www.elijahqi.win/archives/2864
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

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

Sample Input

7 3
0 2 1 2 1 0 0
1 2
3 4
3 5
4 6
5 7
2 5
Sample Output

30
HINT

对于所有数据，1<=n<=100000, 1<=c<=10。

对于15%的数据，n<=2000。
另有5%的数据，所有空地都至多与两个空地相邻。
另有5%的数据，除一块空地与三个空地相邻外，其他空地都分别至多与两个空地相邻。
另有5%的数据，除某两块空地与三个空地相邻外，其他空地都分别至多与两个空地相邻

因为叶子节点只有20个 所以考虑每个叶子节点当根作为trie树的根都是可以生成不同的子串的那么显然我们将这些trie树合并之后统计答案就是我们想要的答案 那么这样一看这个就是一个广义后缀自动机的题了 怎么做 把每个叶子节点当根生成的字符串全部插入一个广义SAM 那么最后查询的时候答案就是len[p]-len[fa[p]] 为什么这么做 因为这是根据了SAM的极简这一条性质

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define N 4000040 
#define N1 110000
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f; 
}
int last=1,root=1,cnt=1,len[N],fa[N],n,C,ch[N][10];
inline int insert1(int x,int p){
    if (ch[p][x]) {
        int q=ch[p][x];
        if(len[p]+1==len[q]) return q;
        int nq=++cnt;memcpy(ch[nq],ch[q],sizeof(ch[q]));len[nq]=len[p]+1;
        fa[nq]=fa[q];fa[q]=nq;for (;p&&ch[p][x]==q;p=fa[p]) ch[p][x]=nq;return nq;
    }
    int np=++cnt;len[np]=len[p]+1;
    for (;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
    if (!p) fa[np]=root;else{
        int q=ch[p][x];if (len[p]+1==len[q]) fa[np]=q;else{
            int nq=++cnt;memcpy(ch[nq],ch[q],sizeof(ch[q]));len[nq]=len[p]+1;
            fa[nq]=fa[q];fa[np]=fa[q]=nq;for (;p&&ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        }
    }return np;
}
struct node{
    int y,next;
}data[N1<<1];long long ans;int h[N1],num,d[N1],cor[N1];
inline void dfs(int x,int f,int r){
    int nxt=insert1(cor[x],r);
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;if(y==f) continue;
        dfs(y,x,nxt);
    }
} 
int main(){
    freopen("bzoj3926.in","r",stdin);
    n=read();C=read();
    for (int i=1;i<=n;++i) cor[i]=read();
    for (int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].y=y;data[num].next=h[x];h[x]=num;
        data[++num].y=x;data[num].next=h[y];h[y]=num;++d[x];++d[y];
    }for (int i=1;i<=n;++i) if (d[i]==1) dfs(i,0,root);
    for (int i=1;i<=cnt;++i) ans+=len[i]-len[fa[i]];
    printf("%lld\n",ans);
    return 0;
}