bzoj 3172 单词 AC自动机

fail 树：一个点只有一个fail指针，所以把fail边倒过来就变成了一个树的结构。一个字符串出现在其他串中，当它是其他串一个后缀的前缀，例如abc出现在aaabcdd中，当它是abcdd这个后缀的前缀。所以需要算一下有多少fail直接或间接指向它最后一个字母c（fail树中 c子孙的大小），就可以算出abc是多少个后缀的前缀，也就可以算出它在整个文本中出现了多少次。
AC自动机建完后，用栈存fail深度，向前递推一遍可以求出。
第一天学，理解很有可能不对…………

#include<cstdio>
#include<iostream>
#include<stack>
#include<queue>
#include<cstring> 
#define SZ 1000005
using namespace std;
struct Trie{
    int ch[SZ][30],tot;
    int d[SZ],val[SZ];
    int lst[SZ],fail[SZ];
    int id[SZ];
    void insert(char *s,int num)
    {
        int p=0;
        int l=strlen(s);
        for(int i=0;i<l;i++)
        {
            int u=s[i]-'a'+1;
            if(!ch[p][u]) ch[p][u]=++tot;
            p=ch[p][u]; d[p]++;
        }
        val[p]++;
        id[num]=p;
    }
    queue<int> q;
    stack<int> S;
    void build()
    {
        for(int i=1;i<=26;i++)
            if(ch[0][i]) q.push(ch[0][i]);
        while(!q.empty())
        {
            int r=q.front();q.pop();
            S.push(r);
            for(int i=1;i<=26;i++)
            {
                int u=ch[r][i];
                if(!u){ch[r][i]=ch[fail[r]][i];continue;}
                q.push(u);
                int v=fail[r];
                while(v&&!ch[v][i])v=fail[v];
                fail[u]=ch[v][i];
                lst[u]=val[fail[u]]?fail[u]:lst[fail[u]];
            }
        }
    }
    void up()
    {
        while(!S.empty())
        {
            int f=S.top();
            S.pop();
            d[fail[f]]+=d[f];
        }
    }
    void print(int i)
    {printf("%d\n",d[id[i]]);}
};
Trie AC;
char s[1000005];
int main()
{
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        AC.insert(s,i);
    }
    AC.build();
    AC.up();
    for(int i=1;i<=n;i++)
        AC.print(i);
}