题意:给你一些字符串,再给你一个长字符串,问之前的那些字符串分别出现过几次,按输入的顺序输出字符串和其出现次数。
跟上题有一点点变化,主要是主串有可能有不是字母,查询时注意一下就可以了。
另开一个数组作为记录每个模式串的次数。
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <malloc.h>
#include <queue>
using namespace std;
int num[1050];
struct Trie
{
Trie *fail,*next[26];
int last;
Trie()
{
for(int i=0;i<26;i++)
next[i]=NULL;
fail=NULL;
last=-1;
}
}*root;
void inserttrie(char *str,int k)
{
int len=strlen(str);
Trie *p=root;
for(int i=0;i<len;i++)
{
int id=str[i]-'A';
if(p->next[id]==NULL)
{
p->next[id]=new Trie;
}
p=p->next[id];
}
p->last=k;
}
void buildfail()
{
Trie *son,*temp,*p=root;
queue<struct Trie*>que;
que.push(p);
while(!que.empty())
{
temp=que.front();
que.pop();
for(int 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 querry(char *str)
{
int len=strlen(str);
Trie *p=root;
Trie *temp;
for(int i=0;i<len;i++)
{
if(str[i]>='A'&&str[i]<='Z')
{
int id=str[i]-'A';
while(p!=root&&p->next[id]==NULL)p=p->fail;
p=p->next[id];
if(!p)p=root;
temp=p;
while(temp!=root)
{
if(temp->last>=0)
{
num[temp->last]++;
}
temp=temp->fail;
}
}
else /*不是大写字母的就回到根节点重新开始*/
{
p=root;
continue;
}
}
}
int main()
{
int n;
char str[1050][55];
char web[2000100];
while(~scanf("%d",&n))
{
memset(num,0,sizeof(num));
root=new Trie;
for(int i=1;i<=n;i++)
{
scanf("%s",str[i]);
inserttrie(str[i],i);
}
buildfail();
scanf("%s",web);
querry(web);
for(int i=1;i<=n;i++)
{
if(num[i]>0)
{
printf("%s: %d\n",str[i],num[i]);
}
}
}
return 0;
}