bzoj3172: [Tjoi2013]单词

最新推荐文章于 2020-01-13 20:23:43 发布

mzajt123

最新推荐文章于 2020-01-13 20:23:43 发布

阅读量446

点赞数 2

CC 4.0 BY-SA版权

分类专栏： bzoj AC自动机文章标签： BZOJ AC自动机 OI

本文链接：https://blog.youkuaiyun.com/mzajt123/article/details/68484611

bzoj 同时被 2 个专栏收录

16 篇文章

订阅专栏

AC自动机

1 篇文章

订阅专栏

本文通过一道编程题目详细介绍了如何使用AC自动机解决字符串匹配问题，包括构建Trie树、计算失败指针以及最终统计词频的具体实现。

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

第一篇竟然是这个题233

刚开始看不懂题呀，然后在Discuss 中才明白题，我语文好菜。

转译下大概就是每一个词，在所有n个词中总共出现多少次。

……词原来可以重复，贡献了2次WA

我的做法：

AC自动机胡写，，，大概在建trie树时记录一下每个节点出现次数，并记录一下重复单词。

然后算fail，然后枚举所有节点让其查找fail直至根，每当遇到单词结尾给这个单词的答案加这个节点的出现次数就好了233

然后大概就好了，，好像描述的……那看代码就好233

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
int n;
int ch[1000005][30];
char s[205][1000005];
int le[205],val[1000005],f[1000005],ti[1000005],to[205];
int cnt=0,ans[205];
queue<int>que;
int main()
{
    int i,j,k,l,u,v;
    scanf("%d",&n);
    for(i=1;i<=n;i++){
        scanf("%s",s[i]+1);
        le[i]=strlen(s[i]+1);
        k=0;
        for(j=1;j<=le[i];j++){
            if(!ch[k][s[i][j]-'a']){cnt++;ch[k][s[i][j]-'a']=cnt;}
            k=ch[k][s[i][j]-'a'];
            ti[k]++;
            }
        to[i]=val[k];
        val[k]=i;
        }
    for(i=0;i<=26;i++)if(ch[0][i])que.push(ch[0][i]);
    while(!que.empty()){
         k=que.front();
         que.pop();
         for(i=0;i<=26;i++){
             if(!ch[k][i]){ch[k][i]=ch[f[k]][i];continue;}
             u=ch[k][i];
             v=f[k];
             que.push(u);
             while(v&&(!ch[v][i]))v=f[v];
             f[u]=ch[v][i];
             }
             }
    for(i=1;i<=cnt;i++){k=i;while(k!=0){ans[val[k]]+=ti[i];k=f[k];}}
    for(i=1;i<=n;i++){k=i;while(to[k]){ans[to[k]]=ans[k];k=to[k];}}
    for(i=1;i<=n;i++)printf("%d\n",ans[i]);
    return 0;
}