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

葉子只有20個，那麽串只有40個
直接上廣義SAM就本質不同子串個數就好了

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXN=100000+10,MAXS=4000000+10;
int n,c,e,beg[MAXN],nex[MAXN<<1],to[MAXN<<1],ch[MAXS<<1][11],len[MAXS<<1],fa[MAXS<<1],las=1,tot=1,col[MAXN],stack[MAXN],cnt,in[MAXN];
ll ans;
template<typename T> inline void read(T &x)
{
    T data=0,w=1;
    char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
    if(ch!='\0')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void insert(int x,int y)
{
    to[++e]=y;
    nex[e]=beg[x];
    beg[x]=e;
}
inline ll extend(int c)
{
    int p=las,np=++tot;
    las=np;
    len[np]=len[p]+1;
    while(p&&!ch[p][c])ch[p][c]=np,p=fa[p];
    if(!p)fa[np]=1;
    else
    {
        int q=ch[p][c];
        if(len[q]==len[p]+1)fa[np]=q;
        else
        {
            int nq=++tot;
            fa[nq]=fa[q];
            memcpy(ch[nq],ch[q],sizeof(ch[nq]));
            len[nq]=len[p]+1;fa[q]=fa[np]=nq;
            while(p&&ch[p][c]==q)ch[p][c]=nq,p=fa[p];
        }
    }
    return (ll)len[np]-len[fa[np]];
}
inline void dfs(int x,int f)
{
    int ch=0;
    stack[++cnt]=col[x];
    for(register int i=beg[x];i;i=nex[i])
        if(to[i]==f)continue;
        else ch++,dfs(to[i],x);
    if(!ch)
    {
        las=1;
        for(register int i=1;i<=cnt;++i)ans+=extend(stack[i]);
    }
    cnt--;
}
int main()
{
    read(n);read(c);
    for(register int i=1;i<=n;++i)read(col[i]);
    for(register int i=1;i<n;++i)
    {
        int u,v;read(u);read(v);
        insert(u,v);insert(v,u);
        in[u]++;in[v]++;
    }
    for(register int i=1;i<=n;++i)
        if(in[i]==1)dfs(i,0);
    write(ans,'\n');
    return 0;
}

【刷題】BZOJ 3926 [Zjoi2015]諸神眷顧的幻想鄉