【HDU 2222】Keywords Search

　　http://acm.hdu.edu.cn/showproblem.php?pid=2222

　　今天学习了一下Aho-Corasick Automation。以前学了trie之后就停止搞字符串这方面了，因为今年NOI出现了AC自动机，所以要学习一下。

　　这个题目的代码很多，我只是贴出来纪念一下第一道AC自动机，我也是跟别人学的。希望做的题多了，我也可以形成自己的风格（这个风格就比较符合我了）……

　　加油！

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
using namespace std;

int n,Cases;
char pattern[51],match[1000001];
struct Aho_Corasick{
	struct NODE{
		int cnt;
		NODE *fail,*next[26];
		NODE(){fail=NULL,cnt=0;memset(next,0,sizeof(next));}
	}*root,*p;
	queue<NODE*> q;
	
	void init(){root=new NODE();}
	
	void build(){
		q.push(root);
		while(!q.empty()){
			NODE *cur=q.front();q.pop();
			for(int i=0;i<26;i++) 
				if(cur->next[i]){
					for(p=cur->fail;p;p=p->fail)
						if(p->next[i]){
							cur->next[i]->fail=p->next[i];
							break;
						}
					if(!p) cur->next[i]->fail=root;
					q.push(cur->next[i]);
				}
		}
	}
	
	void insert(char *s){
		int len=strlen(s);p=root;
		for(int i=0;i<len;i++){
			if(!p->next[s[i]-'a']) p->next[s[i]-'a']=new NODE();
			p=p->next[s[i]-'a'];
		}
		p->cnt++;
	}
	
	void query(char *s){
		int ans=0,len=strlen(s);p=root;
		for(int i=0;i<len;i++){
			while(!p->next[s[i]-'a'] && p!=root) p=p->fail;
			p=p->next[s[i]-'a'];
			if(!p) p=root;
			for(NODE *j=p;j&&j->cnt!=-1;j=j->fail)
				ans+=j->cnt,j->cnt=-1;
		}
		printf("%d\n",ans);
	}
}AC;

int main(){
	scanf("%d",&Cases);
	while(Cases--){
		scanf("%d",&n);
		AC.init();
		while(n--){
			scanf("%s",pattern);
			AC.insert(pattern);
		}
		AC.build();
		scanf("%s",match);
		AC.query(match);
	}
	return 0;
}