BZOJ 3926 ZJOI2015 诸神眷顾的幻想乡 广义后缀自动机

题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3926

 

题意概述:给出一棵N个结点的树,树上最多有20个叶子,每个点有颜色,编号0~9。求树上不同的颜色路径(由一个点到另一个点的路径上的点的颜色按顺序排列而成)数量。N<=100000.

 

当你发现只有20个叶子的时候你就很开心了因为你发现可以乱搞了。

使用秘技——广义后缀自动机来节省空间(实际上广义后缀自动机是对一棵trie做的自动机,解决多母串问题)。可以直接把原树分别以20个结点为根看成20棵trie,弄到SAM里面去。在树的分支处就让这个结点来充当所有儿子的last,仔细一想你发现SAM的性质并没有改变,并且trie的不同分支之间没有影响(说白了广义后缀自动机就是一个能够识别trie中某个结点开始的所有后缀的自动机)。

统计答案的时候直接用每个状态的MAX减去其parent的MAX就可以得到答案。

 

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<cstdlib>
 5 #include<algorithm>
 6 #include<cmath>
 7 #include<queue>
 8 #include<set>
 9 #include<map>
10 #include<vector>
11 #include<cctype>
12 using namespace std;
13 const int MAXN=100005;
14 typedef long long LL;
15 
16 int N,C,c[MAXN];
17 struct edge{ int to,next; }E[MAXN<<1];
18 int first[MAXN],np;
19 struct SAM{
20     static const int maxn=4000005;
21     static const int sigma_sz=10;
22     int sz,to[maxn][sigma_sz],mx[maxn],pa[maxn];
23     SAM(){ sz=1; }
24     int newnode(){
25         memset(to[++sz],0,sizeof(to[sz]));
26         mx[sz]=pa[sz]=0;
27         return sz;
28     }
29     int extend(int w,int p){
30         int np=newnode();
31         mx[np]=mx[p]+1;
32         while(p&&!to[p][w]) to[p][w]=np,p=pa[p];
33         if(!p) pa[np]=1;
34         else{
35             int q=to[p][w];
36             if(mx[q]==mx[p]+1) pa[np]=q;
37             else{
38                 int nq=newnode(); mx[nq]=mx[p]+1;
39                 memcpy(to[nq],to[q],sizeof(to[nq]));
40                 pa[nq]=pa[q];
41                 pa[q]=pa[np]=nq;
42                 while(p&&to[p][w]==q) to[p][w]=nq,p=pa[p];
43             }
44         }
45         return np;
46     }
47     LL calc(){
48         LL re=0;
49         for(int i=2;i<=sz;i++) re+=mx[i]-mx[pa[i]];
50         return re;
51     }
52 }sam;
53 
54 void _scanf(int &x)
55 {
56     x=0;
57     char ch=getchar();
58     while(ch<'0'||ch>'9') ch=getchar();
59     while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
60 }
61 void add_edge(int u,int v)
62 {
63     E[++np]=(edge){v,first[u]};
64     first[u]=np;
65 }
66 void data_in()
67 {
68     _scanf(N);_scanf(C);
69     for(int i=1;i<=N;i++) _scanf(c[i]);
70     int x,y;
71     for(int i=1;i<N;i++){
72         _scanf(x);_scanf(y);
73         add_edge(x,y); add_edge(y,x);
74     }
75 }
76 void DFS(int i,int f,int p)
77 {
78     int pp=sam.extend(c[i],p);
79     for(int p=first[i];p;p=E[p].next){
80         int j=E[p].to;
81         if(j==f) continue;
82         DFS(j,i,pp);
83     }
84 }
85 void work()
86 {
87     for(int i=1;i<=N;i++)
88         if(!E[first[i]].next) DFS(i,0,1);
89     cout<<sam.calc()<<'\n';
90 }
91 int main()
92 {
93     data_in();
94     work();
95     return 0;
96 }

 

posted @ 2018-03-08 17:15  KKKorange  阅读(114)  评论(0编辑  收藏  举报