BZOJ.3926.[ZJOI2015]諸神眷顧的幻想鄉(廣義後綴自動機)

題目連接

要對多個串同時創建SAM，有兩種方法：
1.將全部串拼起來，中間用分隔符隔開，插入字符正常插入便可。
2.在這些串的Trie上建SAM。實際上並不須要建Trie，仍是隻須要正常插入（由於原本就差很少？）。在要插入下一個串時需把las從新設爲root。這就是廣義後綴自動機。

對於本題，由於葉節點最多隻有20個（別理解錯了啊喂），以這些葉節點分別爲根，DFS整棵樹建Trie（固然原圖就是），這樣全部子串就在Trie上某條路徑中。這樣就成了求不一樣子串的個數。
固然仍是不須要建Trie，依次插入SAM便可。若是當前有要插入點的轉移，則再也不新建np，而是直接用p(las)作np。不然會有不少重複節點（雖然不影響正確性吧）。
每次插入一個字符，其產生的子串一共有len[i]個（就是以它爲右端點的後綴），不一樣的子串則有len[i]-len[fa[i]]個。全部節點的貢獻求和即爲答案。

注意會有20次建SAM，空間要夠！還有longlong。

//221660kb  2580ms
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 1000000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
const int N=1e5+7,S=N*20*2;

int n,C,A[N],dgr[N],Enum,H[N],nxt[N<<1],to[N<<1];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Suffix_Automaton
{
    int tot,las,fa[S],son[S][11],len[S];
    void Init(){
        tot=las=1;
    }
    int Insert(int p,int c)
    {
        int las;
        if(son[p][c])
        {
            int q=son[p][c];
            if(len[q]==len[p]+1) las=q;
            else
            {
                int nq=++tot; len[las=nq]=len[p]+1;
                memcpy(son[nq],son[q],sizeof son[q]);
                fa[nq]=fa[q], fa[q]=nq;//不要想固然寫fa[p]=nq q就表明np了 
                for(; son[p][c]==q; p=fa[p]) son[p][c]=nq;
            }
        }
        else
        {
            int np=++tot; len[las=np]=len[p]+1;
            for(; p&&!son[p][c]; p=fa[p]) son[p][c]=np;
            if(!p) fa[np]=1;
            else
            {
                int q=son[p][c];
                if(len[q]==len[p]+1) fa[np]=q;
                else
                {
                    int nq=++tot; len[nq]=len[p]+1;
                    memcpy(son[nq],son[q],sizeof son[q]);
                    fa[nq]=fa[q], fa[np]=fa[q]=nq;
                    for(; son[p][c]==q; p=fa[p]) son[p][c]=nq;
                }
            }
        }
        return las;
    }
    void Calc()
    {
        long long ans=0;
        for(int i=2; i<=tot; ++i) ans+=(long long)(len[i]-len[fa[i]]);
        printf("%lld\n",ans);
    }
}sam;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
inline void AddEdge(int u,int v)
{
    ++dgr[v], to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
    ++dgr[u], to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
void DFS(int x,int f,int rt)
{
    int t=sam.Insert(rt,A[x]);
    for(int i=H[x]; i; i=nxt[i])
        if(to[i]!=f) DFS(to[i],x,t);
}

int main()
{
    n=read(), C=read(), sam.Init();
    for(int i=1; i<=n; ++i) A[i]=read();
    for(int i=1; i<n; ++i) AddEdge(read(),read());
    for(int i=1; i<=n; ++i) if(dgr[i]==1) DFS(i,i,1);
    sam.Calc();

    return 0;
}