【bzoj3926】【Zjoi2015】诸神眷顾的幻想乡

• 题解：

  • 如果树上某个路径串的端点不是叶子，那么一定是另一个串的子串；
  • 这样只对叶子$dfs$把$20*20$个串插入广义$SAM$就是统计本质不同的串的个数的模板了；
  • 我不太会分析广义$SAM$的空间，但是我知道似乎完全图是有$n-1$条欧拉回路的，所以长度不会超过$2n*(20-1)$，需要卡一卡；
  • 可能后面会补广义的坑吧。。。。

  •  #include<bits/stdc++.h>
     #define rg register
     #define il inline
     #define ll long long
     using namespace std;
     const int N=,M=;
     int n,m,sz,mx[N*M],pa[N*M],s[N],len,col[N],ch[N*M][],d[N],cnt,a[N],w[N],st[N*M],o,hd[N];
     ll val[N*M];
     struct Edge{int v,nt;}E[N<<];
     il char gc(){
         static char*p1,*p2,S[];
         if(p1==p2)p2=(p1=S)+fread(S,,,stdin);
         return(p1==p2)?EOF:*p1++;
     }
     il int rd(){
         int x=;char c=gc();
         while(c<''||c>'')c=gc();
         while(c>=''&&c<='')x=(x<<)+(x<<)+c-'',c=gc();
         return x;
     }
     il void adde(int u,int v){
         E[o]=(Edge){v,hd[u]};hd[u]=o++;d[u]++;
         E[o]=(Edge){u,hd[v]};hd[v]=o++;d[v]++;
     }
     il void ins(){
         /*
         for(int i=1;i<=len;i++)printf("%d ",s[i]);
         puts("");
         */
         int lst=,np,nq,p,q,x;
         for(rg int i=;i<=len;i++){
             x=s[i];
             p=lst; mx[lst=np=++sz]=mx[p]+;
             while(p&&!ch[p][x])ch[p][x]=np,p=pa[p];
             if(!p){pa[np]=;continue;}
             q=ch[p][x];
             if(mx[q]==mx[p]+)pa[np]=q;
             else {
                 mx[nq=++sz]=mx[p]+;
                 memcpy(ch[nq],ch[q],sizeof(ch[q]));
                 pa[nq]=pa[q];pa[q]=pa[np]=nq;
                 while(p&&ch[p][x]==q)ch[p][x]=nq,p=pa[p];
             }
         }
     }
     il void dfs(int u,int fa){
         for(rg int i=hd[u];i;i=E[i].nt){
             int v=E[i].v;
             if(v==fa)continue;
             s[++len]=col[v];
             dfs(v,u);
         }
         if(d[u]==&&fa)ins();
         len--;
     }
     int main(){
         #ifndef ONLINE_JUDGE
         freopen("bzoj3926.in","r",stdin);
         freopen("bzoj3926.out","w",stdout);
         #endif
         n=rd();m=rd();
         for(rg int i=;i<=n;i++)col[i]=rd();
         for(rg int i=o=;i<n;i++)adde(rd(),rd());
         sz=;
         for(rg int i=;i<=n;i++)if(d[i]==)a[++cnt]=i;
         for(int i=;i<=cnt;i++){
             s[len=]=col[a[i]];
             dfs(a[i],);
         }
         for(rg int i=;i<=sz;i++)w[mx[i]]++;
         for(rg int i=;i<=n;i++)w[i]+=w[i-];
         for(rg int i=sz;i;i--)st[w[mx[i]]--]=i;
         for(rg int i=sz,u;i;i--){
             val[u=st[i]]=;
             for(rg int j=;j<m;j++){
                 val[u]+=val[ch[u][j]];
             }
         }
         --val[];
         printf("%lld\n",val[]);
         return ;
     }

    bzoj3926