hdu 3065 病毒侵袭持续中 AC自动机

病毒侵袭持续中

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 7489    Accepted Submission(s): 2602

Problem Description

小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下，他发现了网路中的“万恶之源”。这是一个庞大的病毒网站，他有着好多好多的病毒，但是这个网站包含的病毒很奇怪，这些病毒的特征码很短，而且只包含“英文大写字符”。当然小t好想好想为民除害，但是小t从来不打没有准备的战争。知己知彼，百战不殆，小t首先要做的是知道这个病毒网站特征：包含多少不同的病毒，每种病毒出现了多少次。大家能再帮帮他吗？

 

Input

第一行，一个整数N（1<=N<=1000），表示病毒特征码的个数。
接下来N行，每行表示一个病毒特征码，特征码字符串长度在1—50之间，并且只包含“英文大写字符”。任意两个病毒特征码，不会完全相同。
在这之后一行，表示“万恶之源”网站源码，源码字符串长度在2000000之内。字符串中字符都是ASCII码可见字符（不包括回车）。

 

Output

按以下格式每行一个，输出每个病毒出现次数。未出现的病毒不需要输出。
病毒特征码: 出现次数
冒号后有一个空格，按病毒特征码的输入顺序进行输出。

 

Sample Input

  
  

   
   3
AA
BB
CC
ooxxCC%dAAAoen....END
  
  

 

Sample Output

  
  

   
   AA: 2
CC: 1

   
   

    
    

     
     Hint
    
    
Hit：
题目描述中没有被提及的所有情况都应该进行考虑。比如两个病毒特征码可能有相互包含或者有重叠的特征码段。
计数策略也可一定程度上从Sample中推测。

   
   
   
    
  
  

 

注意：这题数据不是很大，但如果不是用数组而是用指针的话，那么字符种类 不能开到128，不然会超内存。可以选择把主串非大写字母 连一起的用‘Z’+1代替。当然写一个del()来回收内存也行。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <limits.h>
#include <malloc.h>
#include <ctype.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <stack>
#include <queue>
#include <vector>
#include <deque>
#include <set>
#include <map> 
 

#define LL long long
#define MIN INT_MIN
#define MAX INT_MAX
#define PI acos(-1.0)
#define FRE freopen("input.txt","r",stdin)
#define FF freopen("output.txt","w",stdout)
#define N 60
#define M 2000010
const int kind = 27;


struct node 
{
    node *fail;//失败指针
    node *next[kind];//Trie每个节点的子节点（最多26个子母）
    int cnt;//是否为该单词的最后一个节点
	int index;
    node () 
	{//构造函数初始化
        fail = NULL;
        cnt = 0;
        memset(next,NULL,sizeof(next));
    };
};
char str[1010][55];//输入的单词
char ss[M];//输入的模式串
queue<node*> qq;//队列构造失败指针
//建Trie树
void BuildTree (node *root,int index) 
{
    node *p = root;
    int i = 0,id;
    while (str[index][i]) 
	{
        id = str[index][i]-'A';
        if (p->next[id] == NULL) 
            p->next[id] = new node();
        p = p->next[id];
        i++;
    }
    (p->cnt)=1;
	(p->index)=index;
}

/*
bfs求失败指针
设这个节点上的字母为C，沿着他父亲的失败指针走，直到走到一个节点，
他的儿子中也有字母为C的节点。然后把当前节点的失败指针指向那个字母也为C的儿子。
如果一直走到了root都没找到，那就把失败指针指向root。
*/
void bfs(node *root) 
{
    while (!qq.empty()) qq.pop();
    int i;
    root->fail = NULL;
    qq.push(root);
	root->fail = NULL;
    node *tmp,*p;
    while (!qq.empty()) 
	{
        tmp = qq.front();
        qq.pop();
        p = NULL;
        for (i = 0; i < kind; i++) 
		{
            if (tmp->next[i] != NULL) 
			{
                p = tmp->fail;
                while (p && !p->next[i]) //p：当前点 tmp不是根  !p->next[i]：还能找到i这个字母 在
                    p = p->fail; 
                if (!p)
					tmp->next[i]->fail = root;
                else 
					tmp->next[i]->fail = p->next[i];
                qq.push(tmp->next[i]);
            }
        }
    }
}
/*
AC自动机主程序
当前字符不匹配，则去当前节点失败指针所指向的字符继续匹配，匹配过程随着指针指向root 或者 匹配到下一个字母 结束。
*/
int ans_id[1010];
int AC_run(node *root) 
{
    int i = 0,ans = 0,id;
    node *p = root;
    while (ss[i])
	{
        id = ss[i] - 'A';
        while (!p->next[id] && p != root)//!p->next[id]  p点下面没有id字母   
            p = p->fail;
        p = p->next[id];
        if(!p) p = root;// while结束 如果上面的p 最后是根的情况下。
        node *tmp = p;
        while (tmp != root) 
		{//cnt=-1 表示已经计算过 
			if(tmp->cnt)
			ans_id[tmp->index]++; 
            tmp = tmp->fail;
        }
        i++;
    }
    return ans;
}

void del(node *p)
{
	for(int i=0;i<kind;i++)
	{
		if(p->next[i]!=NULL)
		{
			del(p->next[i]);
		}
	}
	delete p;
}
int main () 
{
    int t; 
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		node *root = new node();
		memset(ans_id,0,sizeof ans_id);
		for(int i=0;i<n;i++)
		{
			scanf("%s",str[i]);
			BuildTree(root,i);
		}
		bfs(root);
		scanf("%s",ss); 
		int len=strlen(ss);
		int ji=0;
		int on=0;
		for(int i=0;i<len;i++)
		{
			if(!(ss[i]>'Z'||ss[i]<'A'))
			{
				on=1;
				ss[ji++]=ss[i];
			}
			else if(on)
			{
				ss[ji++]='Z'+1;
				on=0;
			}
		}
		ss[ji]=0;
		
		AC_run(root);

		del(root);
		for(int i=0;i<n;i++)
		{
			if(ans_id[i])
			printf("%s: %d\n",str[i],ans_id[i]);
		}
	}
    return 0;
}
/*
9
5
say
shr
she
her
he
*/