HDU 2222 AC自动机模板

最新推荐文章于 2024-04-26 20:01:35 发布

原创最新推荐文章于 2024-04-26 20:01:35 发布 · 487 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#AC自动机

AC自动机专栏收录该内容

2 篇文章

订阅专栏

本文通过一个具体的实战案例，详细介绍了AC自动机的构建过程及其在字符串匹配问题中的应用技巧。文章首先阐述了如何利用AC自动机进行单词树的构建，并在此基础上实现高效的字符串匹配。此外，还特别强调了多组数据处理时的注意事项，如数组清空等细节问题。

题意：给出n个单词，和一篇文章，即一个字符串，求其中有多少个单词出现。
TRICK： ①、重复的单词要分别计算；②、多组数据记得要清空数组，别把不该清的清了，坑死人，最后还是在某位高大佬的帮助下才发现

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<cstring>
#include<string>
#include<algorithm>
#include<iomanip>
#include<iostream>
#include<cctype>
#define mem(x) memset(x,0,sizeof(x)) 
using namespace std;
int t,n,l,len,tot,head=0,tail=1,tt;
char s[1000001],s1[1000001];
int f[500001][27],q[500001],fail[500001],danger[500001],vis[500001];
//-------------------
inline void f1() //建树
{
    int t=0;
    for(int i=1;i<=l;i++)
    {  if(!f[t][s[i]-'a']) 
         f[t][s[i]-'a']=++tot;
       t=f[t][s[i]-'a'];
    }
    danger[t]++;//统计一个单词的出现个数
} 
//-------------------
inline void ac()
{
    head=0,tail=1;
    int j;
    q[1]=0;
    while(head^tail)
    {
        head++;
        j=q[head];
        if(j)
          for(int i=0;i<26;i++)
            if(f[j][i])
            fail[f[j][i]]=f[fail[j]][i]; //求fail指针
        for(int i=0;i<26;i++)
          if(!f[j][i]) f[j][i]=f[fail[j]][i];
          //优化，如果当前的点没有对应的儿子，我们就将它的儿子跳到最终会到的地方
          else q[++tail]=f[j][i]; 
    }
}
//-------------------
int main()
{
    //freopen("keyword.in","r",stdin);
    scanf("%d",&tt);
    for(int bz=1;bz<=tt;bz++)
    {
        mem(f);
        mem(q);
        mem(fail);
        mem(danger);
        mem(vis);
        tot=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s+1);
            l=strlen(s+1);
            f1();
        }
        scanf("%s",s1+1);
        len=strlen(s1+1);
        ac();

        int now=0,tem,ans=0;
        for(int i=1;i<=len;i++)
        {
            now=f[now][s1[i]-'a']；
            tem=now;
            while(tem && vis[tem]!=bz)
            {
                vis[tem]=bz;
                ans+=danger[tem];
                tem=fail[tem];
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}