【ZJOI2015】诸神眷顾的幻想乡

Description

Solution

题意：有一个n个节点的树，每一个节点上有一个字符，定义路径为两点之间的字符组成的字符串，求有多少种不同的路径。

trie上建SAM

原图就是一个trie
很明显可以trie上打一个SAM，但是还是有一些从下到上的路径没有概括到。

依次把所有叶子节点视为根节点

依次把每个叶子节点视为根节点建一棵trie，然后做后缀自动机，这样就可以把每个叶子方向开始的路径概括到。
其实这些trie是可以合并为一个大trie的，然后再见SAM。

其实并不用真的去合并trie

从每个叶子节点搜下去就可以了。

注意空间不要开小了

这可是trie啊！！！

Code

#include<iostream> 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
#define rep(i,a) for(i=first[a];i;i=next[i])
using namespace std;
const int maxn=100007;
int i,j,k,l,n,m;
long long ans;
int a[maxn*2],first[maxn*2],last1[maxn*2],next[maxn*2],tot; 
int last,num,np,nq,p,q,cc[maxn*2],cnt;
struct node{
    int son[12],fa,len;
}t[maxn*20];
void add(int x,int y){
    last1[++tot]=y;next[tot]=first[x];first[x]=tot;
}
int extend(int c,int last){
    np=++num,p=last;
    t[np].len=t[p].len+1;
    while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].fa;
    if(!p)t[np].fa=1;
    else{
        q=t[p].son[c];
        if(t[p].len+1==t[q].len)t[np].fa=q;
        else{
            nq=++num;
            t[nq]=t[q];
            t[nq].len=t[p].len+1;
            t[q].fa=t[np].fa=nq;
            while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].fa;
        }
    }
    last=np;
    return last;
}
void dfs(int x,int y,int z){
    int i,u=extend(a[x],z);
    rep(i,x){
        if(last1[i]!=y){
            dfs(last1[i],x,u);    
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    fo(i,1,n){
        scanf("%d",&a[i]);
    }
    fo(i,1,n-1){
        scanf("%d%d",&k,&l);cc[k]++,cc[l]++;
        add(k,l);add(l,k);
    }
    last=num=1;
    fo(i,1,n){
        if(cc[i]==1){
            dfs(i,0,1);
        }
    }
    fo(i,2,num)ans+=t[i].len-t[t[i].fa].len;
    printf("%lld\n", ans);
}