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

诸神眷顾的幻想乡

Problem 3926. -- [Zjoi2015]诸神眷顾的幻想乡

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

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 2415   Solved: 1417
[ Submit][ Status][ Discuss]

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%的数据,除某两块空地与三个空地相邻外,其他空地都分别至多与两个空地相邻

Source

[ Submit][ Status][ Discuss]

HOME Back

分析

广义SAM的板子题。

首先叶子节点不超过20,那么可以直接对每个叶子节点为根的子树插入到广义SAM中。

因为所有合法的答案一定是某个叶子节点为根的树上的一条链,因此这样可以统计出所有合法的答案

然后就是经典的本质不同子串问题了,\(ans = \sum (len[i] - len[fa[i]])\)

时间复杂度\(O(20*n)\)

代码

#include<bits/stdc++.h>
#define rg register
#define il inline
#define co const
template<class T>il T read(){
    rg T data=0;rg char ch=getchar();
    for(;!isdigit(ch);ch=getchar())if(ch=='-') data=-data;
    for(;isdigit(ch);ch=getchar()) data=data*10+ch-'0';
    return data;
}
template<class T>il T read(rg T&x) {return x=read<T>();}
typedef long long ll;
using namespace std;

co int N=4e6;
int tot=1;
int ch[N][10],fa[N],len[N];
int extend(int p,int c){
    int last;
    if(ch[p][c]){
        int q=ch[p][c];
        if(len[q]==len[p]+1) last=q;
        else{
            int clone=last=++tot;
            memcpy(ch[clone],ch[q],sizeof ch[q]);
            fa[clone]=fa[q],len[clone]=len[p]+1;
            fa[q]=clone;
            for(;ch[p][c]==q;p=fa[p]) ch[p][c]=clone;
        }
    }
    else{
        int cur=last=++tot;
        len[cur]=len[p]+1;
        for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=cur;
        if(!p) fa[cur]=1;
        else{
            int q=ch[p][c];
            if(len[q]==len[p]+1) fa[cur]=q;
            else{
                int clone=++tot;
                memcpy(ch[clone],ch[q],sizeof ch[q]);
                fa[clone]=fa[q],len[clone]=len[p]+1;
                fa[cur]=fa[q]=clone;
                for(;ch[p][c]==q;p=fa[p]) ch[p][c]=clone;
            }
        }
    }
    return last;
}
int n,col[N],deg[N];
vector<int> edg[N];
void dfs(int u,int fa,int last){
    int new_last=extend(last,col[u]);
    for(int i=0,v;i<edg[u].size();++i){
        if((v=edg[u][i])==fa) continue;
        dfs(v,u,new_last);
    }
}
int main(){
    read(n),read<int>();
    for(int i=1;i<=n;++i) read(col[i]);
    for(int i=1,u,v;i<n;++i){
        read(u),read(v);
        edg[u].push_back(v),edg[v].push_back(u);
        ++deg[u],++deg[v];
    }
    for(int i=1;i<=n;++i)
        if(deg[i]==1) dfs(i,0,1);
    ll ans=0;
    for(int i=1;i<=tot;++i) ans+=len[i]-len[fa[i]];
    printf("%lld\n",ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/autoint/p/10885467.html
0条评论
添加一条新回复