bzoj3172: [Tjoi2013]单词（AC自动机）

3172: [Tjoi2013]单词

题目：传送门

题解：

　　 其实这题还是蛮裸的一道AC

　　 关键点在于对失败指针的运用：

　　 把所有的失败指针连在一起其实可以构成一棵树

　　 对于节点i对的fail指向的j，root~j一定在root~i这一段中出现过

　　 那么我们就用小段更新大段，一开始就统计所有以当前节点结尾的串出现的次数

　　 具体看代码

AC代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct Tire
{
    int c[30],s,fail;
    Tire()
    {
        s=fail=0;
        memset(c,-1,sizeof(c));
    }
}tr[511000];
int trlen,ans,n,ed[210];
char a[1110000];
void bt(int id,int root)
{
    int x=root,len=strlen(a+1);
    for(int i=1;i<=len;i++)
    {
        int y=a[i]-'a'+1;
        if(tr[x].c[y]==-1)tr[x].c[y]=++trlen;
        x=tr[x].c[y];
        tr[x].s++;
    }
    ed[id]=x;    
}
int list[1110000];
void bfs()
{
    int head=1,tail=1;list[1]=0;
    while(head<=tail)
    {
        int x=list[head];
        for(int i=1;i<=26;i++)
        {
            int son=tr[x].c[i];
            if(son==-1)continue;
            if(x==0)tr[son].fail=0;
            else
            {
                int j=tr[x].fail;
                while(j!=0 && tr[j].c[i]==-1)j=tr[j].fail;
                tr[son].fail=max(tr[j].c[i],0);
            }
            list[++tail]=son;
        }
        head++;
    }
    for(int i=tail;i;i--)tr[tr[list[i]].fail].s+=tr[list[i]].s;//小段更新大段 
}

int main()
{
    scanf("%d",&n);trlen=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",a+1);
        bt(i,0);
    }
    bfs();
    for(int i=1;i<=n;i++)printf("%d
",tr[ed[i]].s);
    return 0;
}