HDU 2222 ACAM模板(AC自动机)

这里找到了两篇很nice的Trie树（作者Hackbuteer1）以及AC自动机（作者niushuai666）入门详解。博主写的可以说是非常用心了，一看就懂。

题意：给出N（<=10000）个单词（长度<=50）组成的字典。求串T（长度<=1e6）中出现了字典中的多少个单词。

题解：AC自动机裸题。所谓AC自动机就是Trie树加上了KMP的fail指针，而且是多模式的fail指针，意思就是说Trie图是有相同前缀的两个单词会有相同的一段root出发的路径，这个fail指针不再是KMP里某一个串的后缀和前缀相同并取最大，而是当前这个点得到的串的后缀和所有串中某个前缀相同且取最大。求这个fail就只要把KMP的过程改成广搜就行了，因为每个字符下一个状态可以是若干字符。。。每次把T中下一个字符考虑进来，通过fail指针转移到Trie图上的一个点上，然后统计root到这个点得到的串 的所有后缀的cnt（就是通过fail一点一点跳到根，把路径上的答案都统计起来。并把统计过的答案置为-1防止重复计数，同时fail跳到答案是-1的点表示这个点到根的答案已经统计过了，那么就提前停下来。）。

模板是我自己写的。。。可能比较丑。。。马上我会找一个好看的贴过来233

Code：

#include<bits/stdc++.h>
using namespace std;
const int MAX1 = 55;
const int MAX2 = 1000050;
char a[MAX1];
char b[MAX2];
struct ACAM {
	int id;
	int cnt;
	ACAM* fail;
	ACAM* nxt[26];
};
ACAM* root;
void ACAM_Clear(ACAM* root){
	for (int i=0;i<=25;i++){
		if (root->nxt[i]!=NULL){
			ACAM_Clear(root->nxt[i]);
		}
	}
	free(root);
};
ACAM* ACAM_Create(int id){
	ACAM* node = (ACAM*)(malloc(sizeof(ACAM)));
	node->cnt=0;
	node->fail = NULL;
	node->id =id;
	memset(node->nxt,0,sizeof(node->nxt));
	return node;
}
void ACAM_Insert(ACAM* root,char* word){
	ACAM* node = root;
	char* p = word;
	while (*p){
		int id = *p-'a';
		if (node->nxt[id]==NULL){
			node->nxt[id] = ACAM_Create(id);
		}
		node = node->nxt[id];
		p++; 
	}
	node->cnt++;
}
ACAM* que[MAX2*10];
void ACAM_Build(ACAM* root){
	int l=0,r=0;
	for (int i=0;i<=25;i++){
		if (root->nxt[i]!=NULL){
			root->nxt[i]->fail = root;
			for (int j=0;j<=25;j++){
				if (root->nxt[i]->nxt[j]!=NULL){
					root->nxt[i]->nxt[j]->fail = root;
					r++;
					que[r] = root->nxt[i]->nxt[j];
				}
			}
		}
	}
	while (l<r){
		l++;
		ACAM* q = que[l];
//		printf("%d %c\n",l,q->id+'a');
		while (q->fail!=root&&q->fail->nxt[q->id]==NULL){
			q->fail = q->fail->fail;
		}
		if (q->fail->nxt[q->id]!=NULL){
			q->fail = q->fail->nxt[q->id];
		}
//		printf("%d fial = %c\n",l,q->fail->id+'a');
		for (int i=0;i<=25;i++){
			if (q->nxt[i]!=NULL){
				q->nxt[i]->fail = q->fail;
				r++;
				que[r] = q->nxt[i];
			}
		}
	}
}
int ACAM_Search(ACAM* root,char*word){
	int ans=0;
	ACAM* node = root;
	char* p = word;
	while (*p){
//		printf("%c\n",*p);
		int id = *p-'a';
		while (node->nxt[id]==NULL&&node!=root){
			node=node->fail;
		}
		if (node->nxt[id]!=NULL){
			node =  node->nxt[id];
		}
		ACAM* temp = node;
		while (temp!=root&&temp->cnt!=-1){
			ans+=temp->cnt;
			temp->cnt = -1;
			temp=temp->fail;
		}
		p++;
	}
	return ans;
}
void init(){
	ACAM_Clear(root);
	root = ACAM_Create(-1);
}
void input(){
	int n;
	scanf("%d",&n);
	while (n--){
		scanf("%s",a);
		ACAM_Insert(root,a);
	}
}
void solve(){
	ACAM_Build(root);
	scanf("%s",b);
	printf("%d\n",ACAM_Search(root,b));
}
int main(){
	int t;
	scanf("%d",&t);
	root = ACAM_Create(-1);
	while (t--){
		init();
		input();
		solve();
	}
	return 0;
}