HDU3065 病毒侵袭持续中 AC自动机

网址：https://vjudge.net/problem/HDU-3065

题意：

给出$n$个模式串和一个文本串，字符集为可见字符，找出模式串在文本串中的出现次数。

题解：

$AC$自动机的模板题，建出$Trie$图之后跳$fail$指针匹配统计数量然后按题目要求输出即可。

AC代码：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <string>
#include <queue>
using namespace std;
int cnt=0;
int strid=0;
struct node
{
    string str;
    int num;
}s[1005];
struct AC
{
    int trie[50005][130];
    int fail[50005];
    int cntword[50005];
    void insert(string &str)
    {
        int root=0;
        for(int i=0;i<str.size();++i)
        {
            int next=str[i];
            if(!trie[root][next])
                trie[root][next]=++cnt;
            root=trie[root][next];
        }
        cntword[root]=++strid;
    }
    void buildfail()
    {
        queue<int>que;
        int now=0;
        for(int i=0;i<130;++i)
        {
            if(trie[now][i])
            {
                fail[trie[now][i]]=0;
                que.push(trie[now][i]);
            }
        }
        while(!que.empty())
        {
            now=que.front();
            que.pop();
            for(int i=0;i<130;++i)
            {
                if(trie[now][i])
                {
                    fail[trie[now][i]]=trie[fail[now]][i];
                    que.push(trie[now][i]);
                }
                else 
                    trie[now][i]=trie[fail[now]][i];
            }
        }
    }
    void query(string &str)
    {
        int next=0;
        for(int i=0;i<str.size();++i)
        {
            next=trie[next][str[i]];
            for(int j=next;j;j=fail[j])
                ++s[cntword[j]].num;
        }
    }
};
AC ac;
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n;
    string tmp;
    while(cin>>n)
    {
        memset(ac.trie,0,sizeof(ac.trie));
        memset(ac.fail,0,sizeof(ac.fail));
        memset(ac.cntword,0,sizeof(ac.cntword));
        cnt=0,strid=0;
        for(int i=1;i<=n;++i)
        {
            cin>>tmp;
            ac.insert(tmp);
            s[i].str=tmp;
            s[i].num=0;
        }
        ac.buildfail();
        cin>>tmp;
        ac.query(tmp);
        for(int i=1;i<=n;++i)
            if(s[i].num)
                cout<<s[i].str<<": "<<s[i].num<<endl;
    }
    return 0;
}