BZOJ 3172 [Tjoi2013]单词 - ac自动机

大概是fail树的一道裸应用，首先要注意fail指针的特点，即对于每个trie树上节点fail指针指向的节点与根节点形成的串一定是该节点与根节点之间形成的串的最长后缀，而判断一个串在树上总共出现了几次，其数目等于end指针被fail指针指向或间接指向的个数，即统计fail树上关于每个串end节点的子树中有多少元素，由此建立fail数跑一下节点size即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<cstdlib>
#include<algorithm>
 
using namespace std;
 
const int maxn=1001000;
const int maxm=205;
 
int n,cnt;
char s[maxn];
int head[maxn],fail[maxn],num[maxn],pos[maxm],next[maxn][27];
 
struct edge
{
    int to,next;
}e[maxn];
 
void insert(int a,int b)
{
    e[++cnt].to=b;e[cnt].next=head[a];head[a]=cnt;
}
struct node
{
    void trie(int x)
    {
        scanf("%s",s+1);
        int len=strlen(s+1);
        int now=0;
        for(int i=1;i<=len;i++)
        {
            if(!next[now][s[i]-'a'])next[now][s[i]-'a']=++cnt;
            now=next[now][s[i]-'a'];
            num[now]++;
        }
        pos[x]=now;
    }
    void getfail()
    {
        queue<int>q;
        for(int i=0;i<26;i++)if(next[0][i])
            q.push(next[0][i]),insert(0,next[0][i]);
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            for(int i=0;i<26;i++)if(next[u][i])
            {
                int v=fail[u];
                while(v&&!next[v][i])v=fail[v];
                fail[next[u][i]]=next[v][i];
                insert(next[v][i],next[u][i]);
                q.push(next[u][i]);
            }
        }
    }
}acm;
void dfs(int x)
{
    for(int i=head[x];i;i=e[i].next)
    {
        dfs(e[i].to);
        num[x]+=num[e[i].to];
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        acm.trie(i);
    acm.getfail();
    dfs(0);
    for(int i=1;i<=n;i++)
        printf("%d\n",num[pos[i]]);
}