HDU 3065：病毒侵袭持续中

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3065

题目让统计各种模式串再文本串中出现的次数。

AC自动机模板题目，我们只需要将统计时候的标记去掉就可以了，意思就是

统计过的单词还可以再进行统计。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <algorithm>

using namespace std;

const int allSon = 26;     ///只存大写字母
const int maxn = 60000;
int node[maxn][allSon];    ///字典树节点
int fail[maxn];
int num[maxn];
int index[maxn];
char patten[1003][55];     ///存放模式串
char text[2000010];        ///文本串
int ans[maxn];
int id;
///构造字典树
void insertPatten(int identify)
{
    int p = 0;
    int pos = 0;
    int len = strlen(patten[identify]);
    while(pos < len)
    {
        int uppercase = patten[identify][pos]-'A';
        if(node[p][uppercase] == -1)  ///节点不存在，创建节点
        {
            memset(node[id],-1,sizeof(node[id]));
            fail[id] = -1;
            num[id] = 0;
            index[id] = -1;
            node[p][uppercase] = id++;
        }
        p = node[p][uppercase];
        pos++;
    }
    num[p]++;
    index[p] = identify;
}
///找fail指针构造AC自动机
void build_AC_automaton()
{
    int p = 0;
    queue<int>qu;
    qu.push(p);
    while(!qu.empty())
    {
        p = qu.front();
        qu.pop();
        for(int i = 0; i < allSon; i++)
        {
            if(node[p][i] != -1)  ///孩子不是空
            {
                if(p == 0)
                {
                    fail[node[p][i]] = 0;
                }
                else
                {
                    int temp = fail[p];
                    while(temp != -1)
                    {
                        if(node[temp][i] != -1)
                        {
                            fail[node[p][i]] = node[temp][i];
                            break;
                        }
                        temp = fail[temp];
                    }
                    if(temp == -1)
                    {
                        fail[node[p][i]] = 0;
                    }
                }
                qu.push(node[p][i]);
            }
        }
    }
}
void find_in_AC_automaton()
{
    int p = 0;
    int pos = 0;
    while(text[pos]!='\0')
    {
        if(text[pos]>='A'&&text[pos]<='Z')  ///是大写字母
        {
            int uppercase = text[pos]-'A';
            while(node[p][uppercase]==-1 && p!=0)
                p = fail[p];
            p = node[p][uppercase];
            if(p == -1) p = 0;
            int temp = p;
            while(temp!=0) ///这里不需要加任何标记了。
            {
                ans[index[temp]] += num[temp];
                temp = fail[temp];
            }
        }
        else
        {
            p = 0;
        }
        pos++;
    }
}
void init()
{
    memset(node[0],-1,sizeof(node[0]));
    num[0] = 0;
    fail[0] = -1;
    index[0] = -1;
    id = 1;
    memset(ans,0,sizeof(ans));
}
int main()
{
    int N;
    while(~scanf("%d",&N))
    {
        init();
        for(int i = 0; i < N; i++)
        {
            scanf("%s",patten[i]);
            insertPatten(i);
        }
        build_AC_automaton();
        scanf("%s",text);
        find_in_AC_automaton();
        for(int i = 0; i < N; i++)
        {
            if(ans[i])
            {
                printf("%s: %d\n",patten[i],ans[i]);
            }
        }
    }
    return 0;
}