转载自:http://www.cppblog.com/mythit/archive/2009/04/21/80633.html
题意:
t组测试数据,每组有n个单词和一个模式串,问n个单词有多少个在模式串中出现过。
思路:
做的第一道AC自动机模板题。。。
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int kind=26;//结点有26种,a~z
struct node
{
node *fail;//失败指针
node *next[kind];
int count;
node()
{
fail=NULL;
count=0;
memset(next,NULL,sizeof(next));
}
}*q[500001];//队列用于bfs构造失败指针
char keyword[51];//输入的单词
char str[1000001];//模式串
int head,tail;//队列头和尾
void insert(char *str,node *root)
{
node *p=root;
int i=0,index;
while(str[i])
{
index=str[i]-'a';
if(p->next[index]==NULL)
p->next[index]=new node();
p=p->next[index];
i++;
}
p->count++;
}
void build_ac(node *root)
{
int i;
root->fail=NULL;
q[head++]=root;
while(head!=tail)
{
node *temp=q[tail++];
node *p=NULL;
for(int 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,index,len=strlen(str);
node *p=root;
while(str[i])
{
index=str[i]-'a';
while(p->next[index]==NULL&&p!=root)
p=p->fail;
p=p->next[index];
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;
scanf("%d",&t);
while(t--)
{
int n;
head=tail=0;
node *root=new node();
scanf("%d",&n);
getchar();
while(n--)
{
gets(keyword);
insert(keyword,root);
}
build_ac(root);
scanf("%s",str);
//printf("%s\n",str);
printf("%d\n",query(root));
}
}