hdu 2896 病毒侵袭

题目链接：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);
    }
}