标签：
这题应该是用输入的模式串应该是可以有多个相同的，所以有count++，而且有一个地方是
if(temp->count>0) //>0是必须的。
{
cnt+=temp->count;
temp->count=0;
}
temp=temp->fail;
#include<stdio.h>
#include<string.h>
#include<malloc.h>
#include<queue>
using namespace std;
char str[1000000+100];
struct node
{
int count;
struct node *next[26];
struct node *fail;
void init()
{
int i;
for(i=0;i<26;i++)
next[i]=NULL;
count=0;
fail=NULL;
}
}*root;
void insert()
{
int len,k;
node *p=root;
len=strlen(str);
for(k=0;k<len;k++)
{
int pos=str[k]-‘a‘;
if(p->next[pos]==NULL)
{
p->next[pos]=new node;
p->next[pos]->init();
p=p->next[pos];
}
else
p=p->next[pos];
}
p->count++;
}
void getfail()
{
int i;
node *p=root,*son,*temp;
queue<struct node *>que;
que.push(p);
while(!que.empty())
{
temp=que.front();
que.pop();
for(i=0;i<26;i++)
{
son=temp->next[i];
if(son!=NULL)
{
if(temp==root) {son->fail=root;}
else
{
p=temp->fail;
while(p)
{
if(p->next[i])
{
son->fail=p->next[i];
break;
}
p=p->fail;
}
if(!p) son->fail=root;
}
que.push(son);
}
}
}
}
void query()
{
int len,i,cnt=0;
len=strlen(str);
node *p,*temp;
p=root;
for(i=0;i<len;i++)
{
int pos=str[i]-‘a‘;
while(!p->next[pos]&&p!=root) p=p->fail;
p=p->next[pos];
if(!p) p=root;
temp=p;
while(temp!=root)
{
if(temp->count>0)
{
cnt+=temp->count;
temp->count=0;
}
temp=temp->fail;
}
}
printf("%d\n",cnt);
}
int main()
{
int cas,n;
scanf("%d",&cas);
while(cas--)
{
root=new node;
root->init();
root->fail=NULL;
scanf("%d",&n);
int i;
getchar();
for(i=0;i<n;i++)
{
gets(str);
insert();
}
getfail();
gets(str);
query();
}
return 0;
}
标签：
原文地址：http://www.cnblogs.com/jie-dcai/p/4306231.html
