题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2896
题目大意及思路:和前面的一样,只是文章有多个。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 550
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int q[200*Max],cnt,rec;
int ans[5];
int vst[550];
char str[210],s[10100];
struct node
{
int id,fail,flag;
int next[128];
void init()
{
id=fail=flag=0;
memset(next,0,sizeof(next));
}
}tri[200*Max];
void insert(char *s,int id)
{
int i,p,x;
p=0;
for(i=0;s[i];i++)
{
x=s[i];
if(!tri[p].next[x])
{
tri[++cnt].init();
tri[p].next[x]=cnt;
}
p=tri[p].next[x];
}
tri[p].id=id;
}
void bfs()
{
int i,p,suf,head=0,tail=0;
p=0;
for(i=0;i<128;i++)
{
if(tri[0].next[i])
{
q[tail++]=tri[0].next[i];
tri[q[tail-1]].fail=0;
}
else
tri[0].next[i]=0;
}
while(head<tail)
{
p=q[head++];suf=tri[p].fail;
for(i=0;i<128;i++)
{
if(tri[p].next[i])
{
q[tail++]=tri[p].next[i];
tri[q[tail-1]].fail=tri[suf].next[i];
}
else
tri[p].next[i]=tri[suf].next[i];
}
}
}
void query(char *s,int num)
{
int i,p,x,tmp;
p=0;
// memset(vst,0,sizeof(vst));
for(i=0;s[i];i++)
{
x=s[i];
p=tri[p].next[x];
tmp=p;
while(tmp&&tri[tmp].flag!=num)
{
if(tri[tmp].id)
ans[rec++]=tri[tmp].id;
tri[tmp].flag=num;
tmp=tri[tmp].fail;
}
}
}
int main()
{
int i,n,m,total;
while(scanf("%d",&n)!=EOF)
{
cnt=total=0;
tri[0].init();
for(i=1;i<=n;i++)
{
scanf("%s",str);
insert(str,i);
}
bfs();
scanf("%d",&m);
for(i=1;i<=m;i++)
{
rec=0;
scanf("%s",s);
query(s,i);
if(rec==0)
continue;
printf("web %d:",i);
total++;
sort(ans,ans+rec);
for(int j=0;j<rec;j++)
{
printf(" %d",ans[j]);
}
puts("");
}
printf("total: %d\n",total);
}
}