[ZJOI2015]诸神眷顾的幻想乡

OJ题号：
BZOJ3926、洛谷3346

题目大意：
　　给定一棵带权树，定义两个路径是不同的当且仅当两个路径上的权值构成的串不同，求本质不同的路径个数。

思路：
　　广义后缀自动机。
　　从每个叶子结点开始DFS，并依次加入到SAM中，对于每个加入的结点，last状态是其父亲节点所在的状态，其余插入过程与普通SAM类似。
　　每个叶子结点的last状态为root。
　　询问过程与普通SAM完全相同。

#include<cstdio>
#include<cctype>
#include<vector>
#include<cstring>
inline int getint() {
    char ch;
    while(!isdigit(ch=getchar()));
    int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int V=100001;
std::vector<int> e[V];
int deg[V];
inline void add_edge(const int u,const int v) {
    e[u].push_back(v);
    deg[u]++;
}
int col[V];
class GeneralSuffixAutomaton {
    private:
        static const int SIGMA_SIZE=10,MAX_STATE=V<<5;
        struct State {
            int link,go[SIGMA_SIZE],len;
        };
        State s[MAX_STATE];
        int sz,newState(const int l) {
            sz++;
            s[sz].len=l;
            return sz;
        }
        int extend(int p,const int w) {
            int new_p=newState(s[p].len+1);
            while(p&&!s[p].go[w]) {
                s[p].go[w]=new_p;
                p=s[p].link;
            }
            if(!p) {
                s[new_p].link=root;
            } else {
                int q=s[p].go[w];
                if(s[q].len==s[p].len+1) {
                    s[new_p].link=q;
                } else {
                    int new_q=newState(s[p].len+1);
                    memcpy(s[new_q].go,s[q].go,sizeof s[q].go);
                    s[new_q].link=s[q].link;
                    s[q].link=s[new_p].link=new_q;
                    while(p&&s[p].go[w]==q) {
                        s[p].go[w]=new_q;
                        p=s[p].link;
                    }
                }
            }
            return new_p;
        }
    public:
        int root;
        GeneralSuffixAutomaton() {
            root=newState(0);
        }
        void build(const int x,const int par,const int lastptr) {
            int p=extend(lastptr,col[x]);
            for(unsigned i=0;i<e[x].size();i++) {
                int &y=e[x][i];
                if(y==par) continue;
                build(y,x,p);
            }
        }
        long long query() {
            long long ret=0;
            for(int i=1;i<=sz;i++) {
                if(s[i].link) {
                    ret+=s[i].len-s[s[i].link].len;
                }
            }
            return ret;
        }
};
GeneralSuffixAutomaton sam;
int main() {
    int n=getint(),m=getint();
    for(int i=1;i<=n;i++) col[i]=getint();
    for(int i=1;i<n;i++) {
        int u=getint(),v=getint();
        add_edge(u,v);
        add_edge(v,u);
    }
    for(int i=1;i<=n;i++) {
        if(deg[i]==1) {
            sam.build(i,0,sam.root);
        }
    }
    printf("%lld\n",sam.query());
    return 0;
}