注册 登录  
 加关注
   显示下一条  |  关闭
温馨提示!由于新浪微博认证机制调整,您的新浪微博帐号绑定已过期,请重新绑定!立即重新绑定新浪微博》  |  关闭

n+e

NewWeb:http://trinkle.is-programmer.com/

 
 
 

日志

 
 

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

2015-05-14 15:21:17|  分类: BZOJ |  标签: |举报 |字号 订阅

  下载LOFTER 我的照片书  |

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

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 235  Solved: 152
[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%的数据,除某两块空地与三个空地相邻外,其他空地都分别至多与两个空地相邻

Solution

其实就是可持久化后缀自动机。。。加一行搞定

Code

/**************************************************************
    Problem: 3926
    User: wjy1998
    Language: C++
    Result: Accepted
    Time:1832 ms
    Memory:130448 kb
****************************************************************/
 
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1600010
int S=1,tot=1,pos[N],fail[N],len[N],ch[N][10];
int n,c,i,x,y,val[N],et=1,la[N],du[N],q[N],tim,vis[N];long long ans;
struct E{int to,nxt;}e[N];
#define adde(x,y) (du[x]++,e[++et]=(E){y,la[x]},la[x]=et)
int add(int p,int c){
    if(ch[p][c]&&len[p]+1==len[ch[p][c]])return ch[p][c];
    int np=++tot;
    for(len[np]=len[p]+1;p&&!ch[p][c];p=fail[p])ch[p][c]=np;
    if(!p)fail[np]=S;
    else if(len[p]+1==len[ch[p][c]])fail[np]=ch[p][c];
    else{
        int q=ch[p][c],r=++tot;
        fail[r]=fail[q],len[r]=len[p]+1;fail[np]=fail[q]=r;
        memcpy(ch[r],ch[q],sizeof(ch[q]));
        for(;p&&ch[p][c]==q;p=fail[p])ch[p][c]=r;
    }
    return np;
}
void dfs(int x,int fa){
    pos[x]=add(pos[fa],val[x]);
    for(int i=la[x];i;i=e[i].nxt)
    if(e[i].to!=fa)dfs(e[i].to,x);
}
char Ch,B[1<<15],*SS=B,*TT=B;
#define getc() (SS==TT&&(TT=(SS=B)+fread(B,1,1<<15,stdin),SS==TT)?0:*SS++)
int aa;int F(){
    while(Ch=getc(),Ch<'0'||Ch>'9');aa=Ch-'0';
    while(Ch=getc(),Ch>='0'&&Ch<='9')aa=aa*10+Ch-'0';return aa;
}
int main(){
    pos[0]=1;n=F(),c=F();
    for(i=1;i<=n;i++)val[i]=F();
    for(i=1;i<n;i++)x=F(),y=F(),adde(x,y),adde(y,x);
    for(i=1;i<=n;i++)if(du[i]==1)dfs(i,0);
    for(i=tot;i>1;i--)ans+=len[i]-len[fail[i]];
    printf("%lld\n",ans);
}
  评论这张
 
阅读(47)| 评论(0)
推荐 转载

历史上的今天

在LOFTER的更多文章

评论

<#--最新日志,群博日志--> <#--推荐日志--> <#--引用记录--> <#--博主推荐--> <#--随机阅读--> <#--首页推荐--> <#--历史上的今天--> <#--被推荐日志--> <#--上一篇,下一篇--> <#-- 热度 --> <#-- 网易新闻广告 --> <#--右边模块结构--> <#--评论模块结构--> <#--引用模块结构--> <#--博主发起的投票-->
 
 
 
 
 
 
 
 
 
 
 
 
 
 

页脚

网易公司版权所有 ©1997-2018