hdu 2222 Keywords Search（AC自动机入门题）

昨晚开始想学AC自动机，然后尽早看了看算导的KMP。。。以前看过，忘差不多了。

今天看了看一个学习AC自动机的文章，磕磕绊绊理解了。

trie的建立很随意，关键是失败指针这个概念。稍微有点抽象。这个指针的建立，用到了BFS，引用那个文章的话

“  假设有一个节点k，他的失败指针指向j。那么k,j满足这个性质：设root到j的距离为n，则从k之上的第n个节点到k所组成的长度为n的单词，与从root到j所组成的单词相同。对于每个节点，我们可以这样处理：设这个节点上的字母为C，沿着他父亲的失败指针走，直到走到一个节点，他的儿子中也有字母为C的节点。然后把当前节点的失败指针指向那个字目也为C的儿子。如果一直走到了root都没找到，那就把失败指针指向root，最开始，我们把root加入队列(root的失败指针显然指向自己)，这以后我们每处理一个点，就把它的所有儿子加入队列，直到搞完。”

查找匹配数目的时候，如果发现不匹配的就沿着失败指针寻找，直到到根部为止。

因为失败指针走一次后再走就没有用了，可以标记下，下次就不用走了。

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <math.h>
#include <time.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#define MID(x,y) ( ( x + y ) >> 1 )
#define L(x) ( x << 1 )
#define R(x) ( x << 1 | 1 )
#define BUG puts("here!!!")
#define STOP system("pause")

using namespace std;

const int MAX_N = 26;
const int MAX_NODE = 500010;

struct NODE{
	int cnt;
	NODE *next[MAX_N], *fail;
	NODE()
	{
		cnt = 0;
		fail = NULL;
		memset(next, 0, sizeof(next));
	}
}*head;

void Build_trie(char *s,NODE *head)
{
	int len = strlen(s);
	for(int i=0; i<len; i++)
	{
		int k = s[i] - 'a';
		if( head->next[k] == NULL )
			head->next[k] = new NODE();
		head = head->next[k];
	}
	head->cnt++;
}

queue<NODE*> q;
void Build_fail(NODE *head)
{
	head->fail = NULL;
	q.push(head);
	while( !q.empty() )
	{
		NODE *now = q.front(); q.pop();
		for(int i=0; i<MAX_N; i++)
			if( now->next[i] )
			{
				NODE *p = now->fail;
				while( p )
				{
					if( p->next[i] )
					{
						now->next[i]->fail = p->next[i];
						break;
					}
					p = p->fail;
				}
				if( p == NULL )
					now->next[i]->fail = head;
				q.push(now->next[i]);
			}
						
	}
}

int AC_find(NODE *head, char *s)
{
	int len = strlen(s), sum = 0;
	NODE* p = head;
	for(int i=0; i<len; i++)
	{
		int k = s[i] - 'a';
		while( p->next[k] == NULL && p != head )
			p = p->fail;
		p = p->next[k] == NULL ? head : p->next[k];
		
		NODE *tmp = p;
		while( tmp != head && tmp->cnt != -1 )
		{
			sum += tmp->cnt;
			tmp->cnt = -1;
			tmp = tmp->fail;
		}
	}
	return sum;
}

char s[1000005];
char ss[100];
int main()
{
	int n, ncases;

	scanf("%d", &ncases);
	
	while( ncases-- )
	{
		head = new NODE();
		scanf("%d", &n);		
		while( n-- )
		{
			scanf("%s", ss);
			Build_trie(ss, head);
		}
		
		Build_fail( head );
		
		scanf("%s", s);
		int sum = AC_find( head, s);
		printf("%d\n", sum);
	}

return 0;
}