題目描述

幽香是全幻想鄉裏最受人歡迎的萌妹子，這天，是幽香的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個。

SOL：

我們發現，葉子節點異常的少，（膜拜麗姐的語文水平，第一次讀成了每個點的度少於等於20），那麽我們就可以對每一個葉子節點建後綴自動機，在Trie建自動機只要維護當前節點的父親節點在SAM上的點即可。

不會SAM（後綴自動機）的請點這裏：http://blog.csdn.net/qq_35649707/article/details/66473069

不會廣義後綴自動機的請點這裏：http://blog.csdn.net/wangzhen_yu/article/details/45481269

#include<bits/stdc++.h>
#define sight(x) (‘0‘<=x&&x<=‘9‘)
#define N 4000007
using namespace std;
inline void read(int & x){
    static char
 
 c;
    for (c=getchar();!sight(c);c=getchar());
    for (x=0;sight(c);c=getchar()) x=x*10+c-48; 
}
struct S{
    int c[10],fa,val;
}T[N];
int fall[N],net[N],head[N],toT,col[N],n,C,a,b,dre[N];
inline void add(int x,int y) { fall[++toT]=y; net[toT]=head[x]; head[x]=toT;}
int tot=1;
inline int Sam(int x,int last){
    int np=++tot;
    T[np].val=T[last].val+1;
    for(;last&&(!T[last].c[x]);last=T[last].fa) T[last].c[x]=np;
    if (!last) T[np].fa=1;
    else {
        int q=T[last].c[x];
        if (T[last].val+1==T[q].val) T[np].fa=q;
        else {
            int nq=++tot;  T[nq]=T[q];
            T[nq].val=T[last].val+1;
            T[q].fa=T[np].fa=nq;
            for (;last&&T[last].c[x]==q;last=T[last].fa) T[last].c[x]=nq;
        }
    }
    return np;
}
void dfs(int x,int f,int to){
    int g=Sam(col[x],to);
    for (int i=head[x];i;i=net[i])
     if (fall[i]^f) dfs(fall[i],x,g);
}
long long ans;
int main () {
//    freopen("a.in","r",stdin);
    read(n); read(C);
    for (int i=1;i<=n;i++) read(col[i]);
    for (int i=1;i<n;i++){
     read(a); read(b); dre[a]++; dre[b]++; add(a,b); add(b,a);}
    for (int i=1;i<=n;i++)
     if (dre[i]==1) dfs(i,0,1);
    for (int i=1;i<=tot;i++) ans+=T[i].val-T[T[i].fa].val;
    printf("%lld\n",ans);
    return 0;
}

ZJOI 2015 諸神眷顧的幻想鄉