HDU 2222 Keywords Search

ac自动机第一题

代码参考kuangbin，用数组实现

思想参考的这个，写得很详细：http://www.cppblog.com/mythit/archive/2009/04/21/80633.html

#include<stdio.h> 
#include<algorithm>
#include<math.h>
#include<string.h>
#include<queue>
using namespace std;
#define N 500010
struct trie{//其实就是字典树多了个失败指针 
	int next[N][26],fail[N],end[N];
	int root,l;
	int newnode(){
		for(int i=0;i<26;i++) next[l][i]=-1;
		end[l++]=0;
		return l-1;
	}
	int init(){
		l=0;
		root=newnode();
	}
	void insert(char s[]){
		int len=strlen(s);
		int now=root;
		for(int i=0;i<len;i++) {
			if(next[now][s[i]-'a']==-1) next[now][s[i]-'a']=newnode();//如果没出现过，就新建结点
			now=next[now][s[i]-'a'];//找到该结点的下一结点 
		}
		end[now]++;//一个单词完，end+1 
	}
	void build(){//建立失败指针,失败指针是指当前结点的下一字符不匹配时，指向另一结点，再看另一结点的下一字符是否匹配 
		queue<int> q;
		fail[root]=root;//给根结点建失败指针，方便其他结点赋值 
		for(int i=0;i<26;i++){
			if(next[root][i]==-1) next[root][i]=root;//把失配结点的下一结点指向根结点，其实就是失败指针，方便查询时首次赋值: now = next[now][buf[i]-'a']; 
			else {
				fail[next[root][i]]=root;//首字符的失败指针指向根结点
				q.push(next[root][i]);
			}
		}
		while(!q.empty()){
			int now=q.front();
			q.pop();
			for(int i=0;i<26;i++){
				if(next[now][i]==-1) next[now][i]=next[fail[now]][i];//同上 
				else{
					fail[next[now][i]]=next[fail[now]][i];
					q.push(next[now][i]);
				}
			}
		}
	}
	int query(char s[]){
		int len=strlen(s);
		int now=root;
		int ans=0;
		for(int i=0;i<len;i++){
			now=next[now][s[i]-'a'];//注意这里该字符不一定存在，但直接赋值了，所以在build里要把next[][]置为父结点的失败指针
			int tmp=now;
			while(tmp!=root){
				ans+=end[tmp];
				end[tmp]=0;
				tmp=fail[tmp];//用fail找相同前缀的串，其实这里没有匹配fail
			}
		}
		return ans;
	}
	void debug(){//模板里的，方便调试 
		for(int i=0;i<l;i++){
			printf("id = %3d,fail = %3d,end = %3d,chi = [",i,fail[i],end[i]);
            for(int j = 0;j < 26;j++) printf("%2d",next[i][j]);
            printf("]\n");
		}
	}
};
char s[N*2];
trie ac;
int main(){
	/*#ifndef ONLINE_JUDGE
	freopen("in.txt","r",stdin);
	#endif*/
	int t,n;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		ac.init();//每次重新初始化 
		for(int i=0;i<n;i++) {
			scanf("%s",s);
			ac.insert(s);
		}
		ac.build();//建失败指针
		scanf("%s",s);
		printf("%d\n",ac.query(s));
	}
	return 0;
}