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]]);
} 



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值