全全类似kmp般的匹配,需要注意的是 当出现不合法字符的时候 要把p返回到root,之前WA了好几次T T!!
#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
using namespace std;
char x[1002][55];
char a[2000002];
int ans[1002];
struct trie
{
int mark;
trie *fail;
trie *next[27];
trie()
{
mark=-1;
memset(next,0,sizeof(next));
fail=NULL;
}
}*root;
void init(char *v,int id)
{
trie *p=root;
for(int i=0; v[i]; i++)
{
int tep=v[i]-'A';
if(p->next[tep]==NULL) p->next[tep]=new trie();
p=p->next[tep];
}
p->mark=id;
}
void getac()
{
queue<trie*>q;
q.push(root);
while(!q.empty())
{
int i;
trie *p,*tep;
p=q.front();
q.pop();
for(i=0; i<26; i++)
{
if(p->next[i]!=NULL)
{
if(p==root) p->next[i]->fail=root;
else
{
tep=p->fail;
while(tep!=NULL)
{
if(tep->next[i]!=NULL)
{
p->next[i]->fail=tep->next[i];
break;
}
tep=tep->fail;
}
if(tep==NULL) p->next[i]->fail=root;
}
q.push(p->next[i]);
}
}
}
}
void finde()
{
trie *p=root,*q;
for(int i=0; a[i]; i++)
{
if(a[i]<'A'||a[i]>'Z') //注意,发现不合法的要把p返回到root!!
{
p=root;
continue;
}
int tep=a[i]-'A';
while(p->next[tep]==NULL && p!=root) p=p->fail;
p=p->next[tep];
if(p==NULL) p=root;
q=p;
while(q!=root) //这里就类似kmp的匹配了,寻找有多少个!
{
if(q->mark!=-1)
ans[q->mark]++;
q=q->fail;
}
}
}
void del(trie *p)
{
for(int j=0; j<26; j++) if(p->next[j]!=NULL) del(p->next[j]);
free(p);
}
int main()
{
int n;
while(scanf("%d",&n)!=-1)
{
int i;
root=new trie();
for(i=0; i<n; i++)
{
scanf("%s",x[i]);
init(x[i],i);
}
getac();
getchar();
gets(a);
memset(ans,0,sizeof(ans));
finde();
for(i=0; i<n; i++)
{
if(ans[i])
printf("%s: %d\n",x[i],ans[i]);
}
del(root);
}
}