[BZOJ 3172][Tjoi2013]单词(AC自动机)

本文介绍了一种使用AC自动机解决论文中单词计数问题的方法。输入为一系列单词,通过构建AC自动机实现单词匹配与计数,最终输出每个单词在论文中的出现次数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Description


某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。

Input


第一个一个整数N,表示有多少个单词,接下来N行每行一个单词。每个单词由小写字母组成,N<=200,单词长度不超过10^6

Output


输出N个整数,第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input


3
a
aa
aaa

Sample Output


6
3
1

Solution


莫名RE一次0 0
我也不知道我到底改了哪儿就A了(仔细想想,我好像把数组开小了些?什么鬼

其实差不多是裸的AC自动机
insert的时候cnt记录经过该节点的次数
最后逆bfs序把cnt加到fail指针的cnt上

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<queue>
using namespace std;
int n,siz=0,root;
char s[1000005];
struct node{
    int next[26],fail,cnt;
}Trie[1000005];
int id[205],tot=0;
int q[1000005],head,tail;
int newnode()
{
    Trie[++siz].fail=0;
    Trie[siz].cnt=0;
    for(int i=0;i<26;i++)
    Trie[siz].next[i]=0;
    return siz;
}
void _insert(char *word)
{
    int i=0,p=root;
    while(word[i])
    {
        int idx=word[i]-'a';
        if(!Trie[p].next[idx])
        Trie[p].next[idx]=newnode();
        p=Trie[p].next[idx];    
        Trie[p].cnt++;
        i++;
    }
    id[++tot]=p;
}
void build()
{
    head=tail=0;
    q[++tail]=root;
    while(head<tail)
    {
        int p=q[++head];
        for(int i=0;i<26;i++)
        {
            int t=Trie[p].fail;
            while(t&&!Trie[t].next[i])t=Trie[t].fail;
            if(Trie[p].next[i])
            {
                Trie[Trie[p].next[i]].fail=t?Trie[t].next[i]:root;
                q[++tail]=Trie[p].next[i];
            }
            else Trie[p].next[i]=t?Trie[t].next[i]:root;
        }
    }
}
void work()
{
    for(int i=tail;i;i--)
    {
        int t=q[i];
        Trie[Trie[t].fail].cnt+=Trie[t].cnt;
    }
}
int main()
{
    scanf("%d",&n);
    root=newnode();
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        _insert(s);
    }
    build();work();
    for(int i=1;i<=tot;i++)
    {
        printf("%d\n",Trie[id[i]].cnt);
    }
    return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值