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;

}