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

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

---

广义后缀自动机...

久仰公之大名啊...

太阳花田的结构比较特殊，只与一个空地相邻的空地数量不超过$20$个。

这就是说叶子节点的个数不超过$20$个，我们知道树上的每一条路径一定对应了一个:以某一个叶子节点为根的tire树的后缀的前缀(也就是子串),我们知道SAM是可以统计不同子串的个数的即：${\sum _{i=1}^{n} (len[i]-len[fa[i]])}$

考虑如何把这20棵tire构成一个后缀自动机，我们在只有一个串的时候$np$就是直接指向前一个字符的位置，因为我们现在是树结构，所以$np$指向的应该是该点在$trie$树中父节点的位置，其他的就和普通的$SAM$一样，为了找到节点的父亲，$DFS$找即可。

---

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<vector>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 using namespace std;
 #define maxn 1000010
 #define llg long long
 #define SIZE 11
 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
 llg n,m,last=,root=,du[maxn],v[maxn];

 vector<llg>a[maxn];

 struct SAM
 {
     struct
     {
         llg len,f,ch[SIZE];
         void init(){len=,f=,memset(ch,,sizeof(ch));}
     }e[maxn*];

     llg insert(llg p,llg c)
     {
         llg np=++last; e[np].init();
         e[np].len=e[p].len+;
         for (;p && !e[p].ch[c];p=e[p].f) e[p].ch[c]=np;
         if (!p) e[np].f=root;
         else
         {
             llg q=e[p].ch[c];
             if (e[q].len==e[p].len+) e[np].f=q;
             else
             {
                 llg nq=++last;
                 e[nq].init();
                 e[nq]=e[q];
                 e[nq].len=e[p].len+;
                 e[q].f=e[np].f=nq;
                 for (;p && e[p].ch[c]==q;p=e[p].f) e[p].ch[c]=nq;
             }
         }
         return np;
     }
 }sam;

 void init()
 {
     llg x,y;
     cin>>n>>m;
     for (llg i=;i<=n;i++) scanf("%lld",&v[i]);
     for (llg i=;i<n;i++)
     {
         scanf("%lld%lld",&x,&y);
         a[x].push_back(y),a[y].push_back(x);
         du[x]++,du[y]++;
     }
 }

 void dfs(llg x,llg fa,llg p)
 {
     llg t=sam.insert(p,v[x]),w=a[x].size();
     for (llg i=;i<w;i++)
         if (a[x][i]!=fa)
             dfs(a[x][i],x,t);
 }

 int main()
 {
     yyj("god");
     init();
     for (llg i=;i<=n;i++) if (du[i]==) dfs(i,,);
     llg ans=;
     for (llg i=;i<=last;i++) ans+=sam.e[i].len-sam.e[sam.e[i].f].len;
     cout<<ans;
     return ;
 }