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

题目描述

幽香是全幻想乡里最受人欢迎的萌妹子，这天，是幽香的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个。 

输入

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

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

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

输出

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

样例输入

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

样例输出

30

---

题解

广义后缀自动机

“一个点的度数不超过20”=“与同一个空地相邻的空地数量不超过20个”≠“只与一个空地相邻的空地数量不超过20个”=“叶子结点的数量不超过20个”

陈老师语文水平真是高= =

只有20个叶子结点的话，就可以从每个叶子结点开始dfs一遍，对得到的树（很多人把它叫做Trie不明觉厉）构建广义后缀自动机。

然后统计不同字符串个数即可。

统计的方法依旧是原套路：x的出现次数=dis[x]-dis[fa[x]]。

注意需要开long long

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100010
#define M 4000010
using namespace std;
int c[N] , head[N] , to[N << 1] , next[N << 1] , cnt , d[N] , a[M][10] , fa[M] , dis[M] , tot = 1;
void add(int x , int y)
{
    to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
int ins(int p , int c)
{
    int np = ++tot;
    dis[np] = dis[p] + 1;
    while(p && !a[p][c]) a[p][c] = np , p = fa[p];
    if(!p) fa[np] = 1;
    else
    {
        int q = a[p][c];
        if(dis[q] == dis[p] + 1) fa[np] = q;
        else
        {
            int nq = ++tot;
            memcpy(a[nq] , a[q] , sizeof(a[q])) , dis[nq] = dis[p] + 1 , fa[nq] = fa[q] , fa[np] = fa[q] = nq;
            while(p && a[p][c] == q) a[p][c] = nq , p = fa[p];
        }
    }
    return np;
}
void dfs(int x , int fa , int p)
{
    int i , t = ins(p , c[x]);
    for(i = head[x] ; i ; i = next[i]) if(to[i] != fa) dfs(to[i] , x , t);
}
int main()
{
    int n , i , x , y;
    long long ans = 0;
    scanf("%d%*d" , &n);
    for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &c[i]);
    for(i = 1 ; i < n ; i ++ ) scanf("%d%d" , &x , &y) , add(x , y) , add(y , x) , d[x] ++ , d[y] ++ ;
    for(i = 1 ; i <= n ; i ++ ) if(d[i] == 1) dfs(i , 0 , 1);
    for(i = 1 ; i <= tot ; i ++ ) ans += (long long)(dis[i] - dis[fa[i]]);
    printf("%lld\n" , ans);
    return 0;
}