BZOJ 3172: [Tjoi2013]单词(fail树)

题意:
一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

---

思路:
ac自动机的本质如果一个串是另外一个串的子串,那么这个串是另外一个串的前缀的后缀,即另外一个串通过不停地fail指针可以移动
到这个串的末尾

知道了这个理论,那么其实fail指针就是一棵树,求这个串总共出现多少次->这个节点的子树的大小(通过循环传递可以直接求出)

---

#include<bits/stdc++.h>
using namespace std;
char s[1150010];
int ed[1000010],q[1000010];

struct Trie{
    int sz,val[1000010],next[1000010][26],fail[1000010],root;

    int newnode(){
        memset(next[sz],-1,sizeof(next[sz]));
        val[sz]=0,sz++;
        return sz-1;
    }

    void init(){
        sz=0;
        root=newnode();
    }

    void insert(int st,int ed,int &pos){
        int now=root;
        for(int i=st;i<ed;i++){
            if(next[now][s[i]-'a']==-1)
                next[now][s[i]-'a']=newnode();
            now=next[now][s[i]-'a'];
            val[now]++;
        }
        pos=now;
    }

    void build(){
        int head=0,tail=0;
        fail[root]=root;
        for(int i=0;i<26;i++)
            if(next[root][i]!=-1)
                fail[next[root][i]]=root,q[++tail]=next[root][i];
        while(head<tail){
            int now=q[++head];
            for(int i=0;i<26;i++){
                if(next[now][i]!=-1){
                    int k=fail[now];
                    while(k!=root&&next[k][i]==-1)   k=fail[k];
                    if(next[k][i]!=-1)
                        k=next[k][i];
                    fail[next[now][i]]=k;
                    q[++tail]=next[now][i];
                }
            }
        }
        for(int i=tail;i>=1;i--)
            val[fail[q[i]]]+=val[q[i]];
    }
};

Trie ac;
int len[1000];

int main(){
    ac.init();
    int n;
    scanf("%d",&n);
    len[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%s",s+len[i-1]);
        len[i]=strlen(s);
        ac.insert(len[i-1],len[i],ed[i]);
    }
    ac.build();
    for(int i=1;i<=n;i++)
        printf("%d\n",ac.val[ed[i]]);
    return 0;
}