hdu2222 Keywords Search

AC自动机模板题

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
struct node{
	node *fail;
	node *next[26];
	int count;
}*q[500001];
char keyword[51];
char str[1000001];
int head,tail;
void insert(char str[],node *root){
	node *p=root;
	int i=0,cur;
	while(str[i]){
		cur=str[i]-'a';
		if(p->next[cur]==NULL)
			p->next[cur]=new node();
		p=p->next[cur];
		i++;
	}
	p->count++;
}
void build(node *root){
	int i;
	root->fail=NULL;
	q[head++]=root;
	while(head!=tail){
		node *temp=q[tail++];
		node *p=NULL;
		for(i=0;i<26;i++){
			if(temp->next[i]!=NULL){
				if(temp==root)
					temp->next[i]->fail=root;
				else{
					p=temp->fail;
					while(p!=NULL){
						if(p->next[i]!=NULL){
							temp->next[i]->fail=p->next[i];
							break;
						}
						p=p->fail;
					}
					if(p==NULL)
						temp->next[i]->fail=root;
				}
				q[head++]=temp->next[i];
			}
		}
	}
}
int query(node *root){
	int i=0,cnt=0,cur;
	node *p=root;
	while(str[i]){
		cur=str[i]-'a';
		while(p->next[cur]==NULL&&p!=root)
			p=p->fail;
		p=p->next[cur];
		p=(p==NULL)?root:p;
		node *temp=p;
		while(temp!=root&&temp->count!=-1){
			cnt+=temp->count;
			temp->count=-1;
			temp=temp->fail;
		}
		i++;
	}
	return cnt;
}
int main(){
	int t,n;
	scanf("%d",&t);
	while(t--){
		head=tail=0;
		node *root=new node();
		scanf("%d",&n);
		while(n--){
			scanf("%s",keyword);
			insert(keyword,root);
		}
		build(root);
		scanf("%s",str);
		printf("%d\n",query(root));
	}
	return 0;
}