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

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

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1017  Solved: 599
[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]

OTZ 陈老师高深莫测的语文技巧……

通过对本短片小说最后一行的仔细观察，不难看出这块奇葩的田地形成了一棵至多只有20个叶子节点的树（然而我看了半天都没发现）。

所以从这至多20个节点出发，向根DFS，建立广义自动机，那么树上的每一条路径都是一个可被SAM接受的子串，答案即为SAM中不同的子串个数，累加最长路差值即可。

 #include <bits/stdc++.h>

 /* MYSCANNER */

 #define siz 1024

 inline int get_c(void)
 {
     static char buf[siz];
     static char *head = buf + siz;
     static char *tail = buf + siz;

     if (head == tail)
         fread(head = buf, , siz, stdin);

     return *head++;
 }

 inline int get_i(void)
 {
     register int ret = ;
     register int neg = false;
     register int bit = get_c();

     for (; bit < ; bit = get_c())
         if (bit == '-')neg ^= true;

     for (; bit > ; bit = get_c())
         ret = ret *  + bit - '';

     return neg ? -ret : ret;
 }

 const int maxn = ;

 /* AUTOMATON */

 int tail = ;
 int root[maxn];
 int step[maxn];
 int fail[maxn];
 int next[maxn][];

 inline int extend(int p, int c)
 {
     if (next[p][c] && step[next[p][c]] == step[p] + )
         return next[p][c];
     int t = tail++;
     step[t] = step[p] + ;
     while (p && !next[p][c])
         next[p][c] = t, p = fail[p];
     if (p)
     {
         int q = next[p][c];
         if (step[q] == step[p] + )
             fail[t] = q;
         else
         {
             int k = tail++;
             fail[k] = fail[q];
             fail[q] = fail[t] = k;
             step[k] = step[p] + ;
             memcpy(next[k], next[q], sizeof(next[k]));
             while (p && next[p][c] == q)
                 next[p][c] = k, p = fail[p];
         }
     }
     else
         fail[t] = ;
     return t;
 }

 /* GRAPH */

 int n;
 int m;
 int tot;
 int hd[maxn];
 int nt[maxn];
 int to[maxn];
 int cnt[maxn];
 int col[maxn];
 int vis[maxn];
 int que[maxn];

 inline void addEdge(int x, int y)
 {
     nt[++tot] = hd[x]; to[tot] = y; hd[x] = tot;
     nt[++tot] = hd[y]; to[tot] = x; hd[y] = tot;
 }

 void DFS(int u, int f)
 {
     root[u] = extend(root[f], col[u]);
     for (int i = hd[u]; i; i = nt[i])
         if (to[i] != f)DFS(to[i], u);
 }

 /* MAIN FUNC */

 signed main(void)
 {
     n = get_i();
     m = get_i();

     for (int i = ; i <= n; ++i)
         col[i] = get_i();

     for (int i = ; i < n; ++i)
     {
         int x = get_i();
         int y = get_i();
         addEdge(x, y);
         ++cnt[x];
         ++cnt[y];
     }

     root[] = tail++;

     for (int i = ; i <= n; ++i)
         if (cnt[i] == )DFS(i, );

     long long ans = 0LL;

     int head = , tail = ;

     que[tail++] = ;

     while (head != tail)
     {
         int t = que[head++];
         if (t != )
             ans += step[t] - step[fail[t]];
         for (int i = ; i < m; ++i)
             if (next[t][i] && !vis[next[t][i]])
                 vis[next[t][i]] = , que[tail++] = next[t][i];
     }

     printf("%lld\n", ans);
 }

@Author: YouSiki