AC代码:
#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;
char model[1005][55];
char str[2000005],tstr[2000005];
int vis[1005];
struct node
{
node* fail;
node* nnext[26];
int ncount;
node()
{
fail=NULL;
memset(nnext,NULL,sizeof(nnext));
ncount=-1;
}
}* root;
void tire(char* model0,int x)
{
int index,i=0;
node* p=root;
while(model0[i])
{
index=model0[i++]-'A';
if(p->nnext[index]==NULL)
{
p->nnext[index]=new node;
}
p=p->nnext[index];
}
p->ncount=x;
}
void set_fail()
{
queue<node *> q;
root->fail=NULL;
q.push(root);
node* tmp,*p;
while(!q.empty())
{
tmp=q.front();
q.pop();
for(int i=0;i<26;i++)
{
if(tmp->nnext[i]!=NULL)
{
if(tmp==root)
{
tmp->nnext[i]->fail=root;
}
else
{
p=tmp->fail;
while(p!=NULL)
{
if(p->nnext[i]!=NULL)
{
tmp->nnext[i]->fail=p->nnext[i];
break;
}
p=p->fail;
}
if(p==NULL)
{
tmp->nnext[i]->fail=root;
}
}
q.push(tmp->nnext[i]);
}
}
}
}
int find_model(char* str0)
{
node* p=root;
node* tmp;
int cnt=0;
int i=0;
int index;
while(str0[i])
{
index=str0[i++]-'A';
while(p!=root&&p->nnext[index]==NULL)
{
p=p->fail;
}
p=p->nnext[index];
if(p==NULL)
p=root;
tmp=p;
while(tmp!=root)
{
if(tmp->ncount!=-1)
{
vis[tmp->ncount]++;
cnt++;
}
tmp=tmp->fail;
}
}
return cnt;
}
int main()
{
int n1;
while(~scanf("%d",&n1))
{
memset(vis,0,sizeof(vis));
root=new node;
for(int i=0;i<n1;i++)
{
scanf("%s",model[i]);
tire(model[i],i);
}
gets(str);
gets(str);
set_fail();
int len=strlen(str);
int cnt=0;
int flag=0;
for(int i=0;i<len;i++)
{
if(str[i]<='Z'&&str[i]>='A')
{
tstr[flag++]=str[i];
}
else
{
tstr[flag]='\0';
cnt+=find_model(tstr);
flag=0;
}
}
if(flag!=0)
{
tstr[flag]='\0';
cnt+=find_model(tstr);
}
if(cnt!=0)
{
for(int i=0;i<n1;i++)
{
if(vis[i]!=0)
{
printf("%s: %d\n",model[i],vis[i]);
}
}
}
}
return 0;
}
所遇问题及解决办法:
1:首先,此问题是一道多例问题,这点很不容易注意的到。需要在输入时多出读取并求解。
2:空间超出,由于输入串只有大写字母26个,因此,如果每个AC自动机的节点开128个子节点就太大了。所以,把每次输入中的非大写字母忽略,仅读入大写字母部分的串,在进行查找,累计结果即可。