bzoj3926 广义后缀自动机

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=3926
大意：给一棵树，每个节点代表一个字符，任意2个节点之间可以表示为一个字符串，问不同字符串的个数。（数据保证叶子节点<=20）
思路同bzoj2780

先吐槽一下，这3题建边简直要炸了，幸好学到了种新姿势，在结构体里打内置函数（用到外面的变量名 ::变量名），这种题错了完全不想看，幸好没出什么傻逼问题。

回归正题
从叶子节点开始遍历，答案就是遍历出的很多字符串的子串的不同个数（好绕口。。。）。
加到广义后缀自动机中，相同父亲的节点加入时的last也是相同的。（我一开始竟然神经病的将它们加入到trie中，再加到广义后缀自动机中。。。。。。）
统计答案的时候，第一次我是按照以前做的题的思路，先更新len大的，统计它可以走到哪几个不同串。答案就是根节点的值-1；后来%了下神犇的代码。发现只要统计len[x]-len[fa[x]]的值就好了（蒟蒻终究还是太弱了。。。）

%完神犇后的代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,Co,Col[100040],Ru[100050];

struct SAM {
    int son[17],len,fa;
}C[3000050];
int tot,last,V[3000040],q[3000040];
long long sum[3000040],ans;

inline void Add(int x,int &l)
{
    int p=l;
    if (C[p].son[x]) {
        int q=C[p].son[x];
        if (C[p].len+1==C[q].len) l=q;
        else {
            int nq=++tot;   C[nq]=C[q];
            C[nq].len=C[p].len+1;
            C[q].fa=nq;
            for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq; 
            l=nq;
        }
    } else {
        int np=++tot;   l=tot;
        C[np].len=C[p].len+1;
        for (;p&&!C[p].son[x];p=C[p].fa) C[p].son[x]=np;
        if (!p) C[np].fa=1;
        else {
            int q=C[p].son[x];
            if (C[p].len+1==C[q].len) C[np].fa=q;
            else {
                int nq=++tot; C[nq]=C[q];
                C[nq].len=C[p].len+1;
                C[q].fa=C[np].fa=nq;
                for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
            }
        }
    }
}
inline void Pre()
{
    for (int i=1;i<=tot;++i) ++V[C[i].len];
    for (int i=1;i<=tot;++i) V[i]+=V[i-1];
    for (int i=tot;i;--i) q[V[C[i].len]--]=i;
    int x,y;
    for (int i=tot;i;--i) {
        x=q[i]; sum[x]=1;
        for (int j=0;j<Co;++j) 
            if (y=C[x].son[j])
                sum[x]+=sum[y];
    }
//  for (int i=2;i<=tot;++i) ans+=C[i].len-C[C[i].fa].len;
}
struct Trie {
    int son[14];
}T[3000040];
int Tsum;
void Dfs(int x,int l) 
{
        int y,r;
        for (int i=0;i<Co;++i) 
            if (y=T[x].son[i]) {
                r=l;
                Add(i,r);
                Dfs(y,r);
            }
}
struct sef { int ne,en; };
struct Graph {
    sef h[200050];
    int tot,first[100050];
    inline void setl(int x,int y) {
        h[++tot].ne=first[x]; h[tot].en=y;
        first[x]=tot;
    }
    void Dfs(int x,int fa,int rt) {
        int y;
        if (!T[rt].son[Col[x]]) T[rt].son[Col[x]]=++Tsum;
        int ret=T[rt].son[Col[x]];
        for (int i=first[x];y=h[i].en,i;i=h[i].ne) 
            if (y!=fa)  Dfs(y,x,ret);
    }
}G;
int main()
{
    scanf("%d%d",&n,&Co);
    for (int i=1;i<=n;++i) scanf("%d",Col+i);
    int x,y;
    for (int i=1;i<n;++i) {
        scanf("%d%d",&x,&y);
        ++Ru[x]; ++Ru[y];
        G.setl(x,y);
        G.setl(y,x);
    }
    for (int i=1;i<=n;++i) 
        if (Ru[i]==1) G.Dfs(i,0,0);
    tot=last=1;
    Dfs(0,1);
    Pre();
    printf("%lld\n",sum[1]-1);
}

加到trie中的傻逼代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,Co,Col[100040],Ru[100050];

struct SAM {
    int son[17],len,fa;
}C[3000050];
int tot,last,V[3000040],q[3000040];
long long sum[3000040],ans;

inline void Add(int x,int &l)
{
    int p=l;
    if (C[p].son[x]) {
        int q=C[p].son[x];
        if (C[p].len+1==C[q].len) l=q;
        else {
            int nq=++tot;   C[nq]=C[q];
            C[nq].len=C[p].len+1;
            C[q].fa=nq;
            for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq; 
            l=nq;
        }
    } else {
        int np=++tot;   l=tot;
        C[np].len=C[p].len+1;
        for (;p&&!C[p].son[x];p=C[p].fa) C[p].son[x]=np;
        if (!p) C[np].fa=1;
        else {
            int q=C[p].son[x];
            if (C[p].len+1==C[q].len) C[np].fa=q;
            else {
                int nq=++tot; C[nq]=C[q];
                C[nq].len=C[p].len+1;
                C[q].fa=C[np].fa=nq;
                for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
            }
        }
    }
}
inline void Pre()
{
    for (int i=1;i<=tot;++i) ++V[C[i].len];
    for (int i=1;i<=tot;++i) V[i]+=V[i-1];
    for (int i=tot;i;--i) q[V[C[i].len]--]=i;
    int x,y;
    for (int i=tot;i;--i) {
        x=q[i]; sum[x]=1;
        for (int j=0;j<Co;++j) 
            if (y=C[x].son[j])
                sum[x]+=sum[y];
    }
//  for (int i=2;i<=tot;++i) ans+=C[i].len-C[C[i].fa].len;
}
struct Trie {
    int son[14];
}T[3000040];
int Tsum;
void Dfs(int x,int l) 
{
        int y,r;
        for (int i=0;i<Co;++i) 
            if (y=T[x].son[i]) {
                r=l;
                Add(i,r);
                Dfs(y,r);
            }
}
struct sef { int ne,en; };
struct Graph {
    sef h[200050];
    int tot,first[100050];
    inline void setl(int x,int y) {
        h[++tot].ne=first[x]; h[tot].en=y;
        first[x]=tot;
    }
    void Dfs(int x,int fa,int rt) {
        int y;
        if (!T[rt].son[Col[x]]) T[rt].son[Col[x]]=++Tsum;
        int ret=T[rt].son[Col[x]];
        for (int i=first[x];y=h[i].en,i;i=h[i].ne) 
            if (y!=fa)  Dfs(y,x,ret);
    }
}G;
int main()
{
    scanf("%d%d",&n,&Co);
    for (int i=1;i<=n;++i) scanf("%d",Col+i);
    int x,y;
    for (int i=1;i<n;++i) {
        scanf("%d%d",&x,&y);
        ++Ru[x]; ++Ru[y];
        G.setl(x,y);
        G.setl(y,x);
    }
    for (int i=1;i<=n;++i) 
        if (Ru[i]==1) G.Dfs(i,0,0);
    tot=last=1;
    Dfs(0,1);
    Pre();
    printf("%lld\n",sum[1]-1);
}