病毒侵袭持续中

Problem‘s Link：http://acm.hdu.edu.cn/showproblem.php?pid=3065

中文题，不解释。

analyse:

AC自动机的运用。这一题需要将模式串都存储下来，还有就是base的取值一定要弄清楚，由于这题的模式串都是大写字母所以我们可以通过剪枝来加速。

Time complexity：o(n)+o(ml) 

Source code：

// Memory   Time
// 1347K     0MS
// by : Snarl_jsb
// 2014-09-30-21.00
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<climits>
#include<cmath>
#define LL long long
using namespace std;

char backup[1002][53];
int res[1002];
const int N = 1010;
char str[2000010];
struct node
{
    node *next[26];     //  每个结点都对应26个字母的指针
    node *fail;     //      失配指针
    int count;      //
    int num;
    node()      //  构造函数初始化
    {
        for(int i = 0; i < 26; i++)
            next[i] = NULL;
        count = 0;
        num=0;
        fail = NULL;
    }
}*q[50*N];
node *root;
int head, tail;

void Insert(char *str,int num) //   插入单词.相当于构建一个Trie树
{
    node *p = root;
    int i = 0, index;
    while(str[i])
    {
        index = str[i] - ‘A‘; //  转化为相对数字来存
        if(p->next[index] == NULL) // 该字母未插入过
            p->next[index] = new node();    //  为该字母申请一个结点
        p = p->next[index];     //   移至下一个
        i++;
    }
    p->count++;     //      记录该结点的单词总共插入的次数
    p->num=num;
}
void build_ac_automation(node *root)        //      bfs建立fail指针
{
    root->fail = NULL;
    q[tail++] = root;
    while(head < tail) {
        node *temp = q[head++];
        node *p = NULL;
        for(int i = 0; i < 26; i++) {
            if(temp->next[i] != NULL) {
                if(temp == root) temp->next[i]->fail = root;
                else {
                    p = temp->fail;
                    while(p != NULL) {
                        if(p->next[i] != NULL) {
                            temp->next[i]->fail = p->next[i];
                            break;
                        }
                        p = p->fail;
                    }
                    if(p == NULL) temp->next[i]->fail = root;
                }
                q[tail++] = temp->next[i];
            }
        }
    }
}

int Query(node *root)       //  匹配 + 统计
{
    int i = 0, cnt = 0, index;
    node *p = root;
    while(str[i])
    {
        index = str[i] - ‘A‘;
        if(index<0||index>25)   ///这个地方要特别注意，由于病毒只包含大写字母，所以这儿需要剪枝，不剪枝的话其他地方加判断也可以过
		{
			p=root;
			i++;
			continue;
		}
        while(p->next[index] == NULL && p != root) //前缀是相同的，所以不管哪个指针走到了count不为0的结点上，那么该结点所代表的单词就匹配成功
            p = p->fail;//失配情况下，p指针指向p->fail.(相当于KMP的next数组)
        p = p->next[index];//由于现在所在的位置是父节点，所以需要向下移动一个位置
        if(p == NULL)
            p = root; //如果匹配失败，移动到root，重新开始匹配
        node *temp = p;//
        while(temp != root && temp->count>0)  //统计--如果匹配成功,那么count>1，表示该结点代表的单词数量;否则表示该结点没有单词
        {
//            cnt += temp->count; //统计该单词出现的次数
            res[temp->num]++;   //每次回溯都会加1
//            temp->count = -1;   //!!!!!!!!!!!!!!!!!(如果要重复统计，请讲这句去掉)!!!!!!!!标记为-1，表示该单词已经加入了cnt中
            temp = temp->fail;//判断整条链上的匹配情况
        }
        i++;
    }
    return cnt;
}

int main()
{
    int n,m;
    while(cin>>n)
    {
        head = tail = 0;    //  清零
        root = new node();      //  申请新的root结点
        memset(backup,0,sizeof(backup));
        memset(res,0,sizeof(res));
        for(int i=1;i<=n;++i)
        {
            scanf("%s",str);
            strcpy(backup[i],str);
            Insert(str,i);
        }
        build_ac_automation(root);
        scanf("%s",str);
        Query(root);
        for(int i=1;i<=n;++i)
        {
            if(res[i])
            {
                printf("%s: %d\n",backup[i],res[i]);
            }
        }
    }
    return 0;
}