bzoj 3926 [Zjoi2015]诸神眷顾的幻想乡

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个表示这个树上的叶子不超过20个。。。我原来以为是度小于20。。。

这样的话发现我们枚举每一个叶子作为根，这样的话根到叶子的路径最多就只有400个。

把这些串都建到一个广义后缀自动机里，因为是求有多少个不同的串，那就是$\sum dis[i]-dis[fa[i]]$

#include <cstdio>
#include <cstring>
#define N 100010
using namespace std;
int inline read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int dis[N*40],fa[N*40],ch[N*40][11],sz=1,root=1,last=root;
void insert(int x)
{
    int now=ch[last][x],pre=last;
    if (now)
        if (dis[now]==dis[pre]+1) last=now;
        else
        {
            int nows=++sz;dis[nows]=dis[pre]+1;
            memcpy(ch[nows],ch[now],sizeof ch[now]);
            fa[nows]=fa[now],fa[now]=nows;
            for (;pre && ch[pre][x]==now;pre=fa[pre]) ch[pre][x]=nows;
            last=nows;
        }
    else
    {
        now=++sz,dis[now]=dis[pre]+1;last=now;
        for (;pre && !ch[pre][x];pre=fa[pre]) ch[pre][x]=now;
        if (!pre) fa[now]=root;
        else if (dis[ch[pre][x]]==dis[pre]+1) fa[now]=ch[pre][x];
        else
        {
            int q=ch[pre][x],nows=++sz;dis[nows]=dis[pre]+1;
            memcpy(ch[nows],ch[q],sizeof ch[nows]);
            fa[nows]=fa[q],fa[q]=fa[now]=nows;
            for (;pre && ch[pre][x]==q;pre=fa[pre]) ch[pre][x]=nows;
        }
    }
}
void getans()
{
    long long ans=0;
    for (int i=1;i<=sz;i++)
        ans+=dis[i]-dis[fa[i]];
    printf("%lld\n",ans);
}
struct edge
{
    int to,next;
}e[N*2];
int st[N],tot,v[N],d[N];
void add(int x,int y)
{
    e[++tot].next=st[x];
    e[tot].to=y,st[x]=tot;
}
void dfs(int now,int pre)
{
    insert(v[now]);
    int te=last;
    for (int i=st[now];i;i=e[i].next)
        if (e[i].to!=pre)
            dfs(e[i].to,now),last=te;
}
int n,m,x,y;
main()
{
    n=read(),m=read();
    for (int i=1;i<=n;i++)
        v[i]=read();
    for (int i=1;i<n;i++)
        x=read(),y=read(),
        add(x,y),add(y,x),d[y]++,d[x]++;
    for (int i=1;i<=n;i++) if (d[i]==1)
        last=root,dfs(i,-1);
    getans();
}