【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=100010,M=40;
    int n,m,sz,mx[N*M],pa[N*M],s[N],len,col[N],ch[N*M][10],d[N],cnt,a[N],w[N],st[N*M],o,hd[N];
    ll val[N*M];
    struct Edge{int v,nt;}E[N<<1];
    il char gc(){
        static char*p1,*p2,S[1000000];
        if(p1==p2)p2=(p1=S)+fread(S,1,1000000,stdin);
        return(p1==p2)?EOF:*p1++;
    } 
    il int rd(){
        int x=0;char c=gc();
        while(c<'0'||c>'9')c=gc();
        while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+c-'0',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=1,np,nq,p,q,x;
        for(rg int i=1;i<=len;i++){
            x=s[i];
            p=lst; mx[lst=np=++sz]=mx[p]+1;
            while(p&&!ch[p][x])ch[p][x]=np,p=pa[p];
            if(!p){pa[np]=1;continue;}
            q=ch[p][x];
            if(mx[q]==mx[p]+1)pa[np]=q;
            else {
                mx[nq=++sz]=mx[p]+1;
                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]==1&&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=1;i<=n;i++)col[i]=rd();
        for(rg int i=o=1;i<n;i++)adde(rd(),rd());
        sz=1;
        for(rg int i=1;i<=n;i++)if(d[i]==1)a[++cnt]=i;
        for(int i=1;i<=cnt;i++){
            s[len=1]=col[a[i]];
            dfs(a[i],0);
        }
        for(rg int i=1;i<=sz;i++)w[mx[i]]++;
        for(rg int i=1;i<=n;i++)w[i]+=w[i-1];
        for(rg int i=sz;i;i--)st[w[mx[i]]--]=i;
        for(rg int i=sz,u;i;i--){
            val[u=st[i]]=1;
            for(rg int j=0;j<m;j++){
                val[u]+=val[ch[u][j]];
            }
        }
        --val[1];
        printf("%lld\n",val[1]);
        return 0;
    }