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

题目链接

bzoj3926

题目描述

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

题解

传统后缀自动机是解决单个主串的匹配问题，广义后缀自动机可以用来解决多个主串的匹配问题。
如何将多个主串构建成广义后缀自动机？先将一个主串建立成后缀自动机，让后将重置last，令last=root，下一个字符串再从头节点开始建立，下一状态如果存在且step=last的step+1则直接返回这个节点，否则以后缀自动机的规则进行建立新节点。
这题从叶子节点dfs整棵树，相当于是一颗trie，建出广义后缀自动机，然后像普通的后缀自动机统计不同子串就行了。
建后缀自动机是我们可以令cur[i]表示建到树上i号节点时后缀自动机的last，考虑儿子时直接从父亲的cur开始。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=100010;
int go[N*20][11],pre[N*20],step[N*20],c[N],first[N],d[N],cur[N],tot,n,m,cnt,x,y;
long long ans;
struct edge{
    int x,next;
}e[N*2];

void add(int x,int y){
    e[++tot].x=y;
    e[tot].next=first[x];
    first[x]=tot;
}
int ins(int p,int c){
    int np=go[p][c];
    if(np&&step[np]==step[p]+1) return np;
    np=++cnt; step[np]=step[p]+1;
    for(;p&&go[p][c]==0;p=pre[p]) go[p][c]=np;
    if(p==0) return pre[np]=1,np;
    int q=go[p][c];
    if(step[q]==step[p]+1) return pre[np]=q,np;
    int nq=++cnt; 
    step[nq]=step[p]+1; pre[nq]=pre[q];
    memcpy(go[nq],go[q],sizeof(go[q]));
    pre[q]=pre[np]=nq;
    for(;go[p][c]==q;p=pre[p]) go[p][c]=nq;
    return np;
}
void dfs(int x,int y){
    cur[x]=ins(cur[y],c[x]);
    for(int i=first[x];i;i=e[i].next)
    if(e[i].x!=y) dfs(e[i].x,x);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&c[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        d[x]++; d[y]++;
        add(x,y); add(y,x);
    }
    cur[0]=cnt=1;
    for(int i=1;i<=n;i++)
    if(d[i]==1) dfs(i,0);
    for(int i=2;i<=cnt;i++) 
    ans+=step[i]-step[pre[i]];
    printf("%lld\n",ans);
}