HDU 2222 Keywords Search

题目传送门：http://acm.hdu.edu.cn/showproblem.php?pid=2222

题意：给定一个字符串和n个单词，问字符串中共出现了多少个单词。

解析：这是AC自动机最基础的题目，也是我接触的第一个AC自动机的题目。AC自动机的话，大家可以去参考一下http://blog.csdn.net/niushuai666/article/details/7002823这篇博客，我感觉讲的很详细。然后基本上大致编出来AC自动机没问题。

但是有些需要注意的地方

1.在储存节点是否为单词节点时（我是用val，上面的链接中用的是count）一定要注意出现的单词可能重复，所以必然val是++，

2.不能直接memset(ch,0,sizeof(ch))，这样的话会MLE，虽说不知道为什么╮(╯▽╰)╭

 

#include"stdio.h"
#include"string.h"
#include"queue"
using namespace std;
#define maxnode (500005)
#define sigma_size 26

struct AC_automation{			
	struct node{
		int next[sigma_size];
		int val;	//树节点的附加信息
		int fail;	//存储节点的失配指针
		void init(){
			memset(next,0,sizeof(next));
			fail = val = 0;
		}
	}ch[maxnode];	//用来存储Trie树
	int sz;							//整棵树节点的总数 
	int idx(char c){
		return c - 'a';		//对于小写字母集,获得c的编号
	}

	void init(){
		sz = 1;
		ch[0].init();
	//	memset(ch,0,sizeof(ch)); //虽然不知道为什么，但是加上这句就MLE
	}
	void insert(char *s){
		int u = 0,len = strlen(s);
		for(int i = 0 ; i < len ; i ++){
			int c = idx(s[i]);
			if(ch[u].next[c] == 0){		
				ch[sz].init();
				ch[u].next[c] = sz++;	
			}
			u = ch[u].next[c];
		}
		ch[u].val++;		//此处不可以是ch[u].val = 1;因为可能有多个相同的单词
	}

	//利用bfs得到失配指针fail
	void get_fail(){
		queue<int>Q;
		for(int i = 0 ; i < sigma_size; i ++)
			if(ch[0].next[i] != 0)
				Q.push(ch[0].next[i]);
		while(!Q.empty()){
			int temp = Q.front();
			Q.pop();
			for(int i = 0 ; i < sigma_size; i ++){
				if(ch[temp].next[i] != 0){
					Q.push(ch[temp].next[i]);
					int fail_temp = ch[temp].fail;
					//关键步骤，如果fail指针所指的节点没有('a'+i)这个子节点，继续递归，直到fail_temp指向0节点，即根节点
					while(fail_temp > 0 && ch[fail_temp].next[i] == 0) 
						fail_temp = ch[fail_temp].fail;

					if(ch[fail_temp].next[i] != 0)	//如果fail_temp节点的子节点有('a'+i)这个字符
							fail_temp = ch[fail_temp].next[i];
					ch[ch[temp].next[i]].fail = fail_temp;   //子节点的失配节点由父节点节点决定
				}
			}
		}
	}

	//寻找text[]总共有多少个单词
	int get_word_num(char *text){
		int ans = 0;
		int len = strlen(text);
		int node = 0,fail_temp = 0;
		for(int i = 0 ; i < len ; i ++){
			int character = idx(text[i]);
			while(fail_temp > 0 && ch[fail_temp].next[character] == 0)	
				//若没有text[i]这个字符，则使用失配指针继续在Trie树上遍历
					fail_temp = ch[fail_temp].fail;

			if(ch[fail_temp].next[character] != 0){	//此时找到了text[i]这个字符
				fail_temp = ch[fail_temp].next[character];
				int fail_temp2 = fail_temp;			//此处必须有另一个变量来记录失配的时候的下标
				while(fail_temp2 > 0 && ch[fail_temp2].val){   //在向上遍历失配函数的时候，只有当该节点是单词节点，才会继续失配
					ans += ch[fail_temp2].val;
					ch[fail_temp2].val = 0;					//防止重复计算，把该单词节点删去
					fail_temp2 = ch[fail_temp2].fail;	
				}
			}
		}
		return ans;
	}
}AC;
int N;
char text[1000010];

int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&N);
		char str_temp[55];
		AC.init();
		for(int i= 0 ; i < N; i ++){
			scanf("%s",str_temp);
			AC.insert(str_temp);
			memset(str_temp,0,sizeof(str_temp));
		}
		AC.get_fail();
		scanf("%s",text);
		printf("%d\n",AC.get_word_num(text));
	}
	return 0;
}