病毒侵袭持续中 HDU - 3065 （AC自动机统计模式串出现次数）

病毒侵袭持续中 HDU - 3065

题意：给你n个模式串，1个文本串。统计各个模式串在文本串中出现的次数。
解题思路：
将模式串添加进入字典树时，纪录每个模式串在字典树上终点的位置。对文本串进行匹配时，每匹配到一个模式串，那个模式串出现的数量增加1。
坑点:

1. 多组输入。没写多组输入的话返回的是wa
2. 初始化模式串在字典树上终点的位置。虽然说后面的将模式串添加进字典树时，会自动覆盖。但是有可能部分没有被覆盖的值会在进行文本串匹配时出现bug。如果没有初始化这个数组会返回MLE而不是WA。因为答案可能会溢出，于是变成了负数，所以返回的是MLE。

---

AC代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
//#pragma GCC optimize(3,"Ofast","inline")
#define LL long long
#define MT(a,b) memset(a,b,sizeof(a))
const int mod=1000007;
const int maxn=2e6+10;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
char t[1005][55],s[maxn];
int trie[50005][26],id[50005],fail[50005],vis[1005],tot;
void add_trie(char *str,int n){
    int len=strlen(str);
    int pos=0;
    for (int i=0;i<len;i++){
        if (!trie[pos][str[i]-'A']) trie[pos][str[i]-'A']=++tot;
        pos=trie[pos][str[i]-'A'];
    }
    id[pos]=n;
}
void get_fail(){
    queue<int>q;
    for (int i=0;i<26;i++){
        if (trie[0][i]) {
            q.push(trie[0][i]);
            fail[trie[0][i]]=0;
        }
    }
    while (!q.empty()){
        int pos=q.front();
        q.pop();
        for (int i=0;i<26;i++){
            if (trie[pos][i]){
                fail[trie[pos][i]]=trie[fail[pos]][i];
                q.push(trie[pos][i]);
            } else trie[pos][i]=trie[fail[pos]][i];
        }
    }
}
void build_trie(int n){
    MT(trie,0);
    MT(vis,0);
    MT(id,0);
    tot=0;
    for (int i=1;i<=n;i++){
        scanf("%s",t[i]);
        add_trie(t[i],i);
    }
    get_fail();
}
void solve(char *str){
    int len=strlen(str);
    for (int i=0,pos=0;i<len;i++){
        int k=str[i]-'A';
        if (k<0||k>25) {
            pos=0;
            continue;
        }
        pos=trie[pos][k];
        for (int j=pos;j;j=fail[j]){
            vis[id[j]]++;
        }
    }
}
int main(){
    int n;
    while (~scanf("%d",&n)){
        build_trie(n);
        scanf("%s",s);
        solve(s);
        for (int i=1;i<=n;i++){
            if (!vis[i]) continue;
            printf("%s",t[i]);
            printf(": %d\n",vis[i]);
        }
    }
    return 0;
}