杭电2896(HDU2896)

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100010
struct node
{
    node *fail;
    node *next[94];
    int count,id;
    node()
    {
        fail=NULL;
        count=0;
        for(int i=0;i<94;i++)
            next[i]=NULL;
    }
}*q[N*6];
node *root;
char s[N],str[220];
int vis[550];//标记数组,用来判断该字符串是否出现
void insert(char *s,int c)
{
    int i,j,len=strlen(s);
    node *p=root;
    for(i=0;i<len;i++)
    {
        j=s[i]-32;
        if(p->next[j]==NULL)
            p->next[j]=new node();
        p=p->next[j];
    }
    p->count++;
    p->id=c+1;
}
void build()
{
    int i,j,head=0,tail=0;
    node *p,*tmp;
    q[tail++]=root;
    while(head!=tail)
    {
        p=q[head++];
        for(i=0;i<94;i++)
        {
            if(p->next[i]!=NULL)
            {
                if(p==root)
                    p->next[i]->fail=root;
                else
                {
                    tmp=p->fail;
                    while(tmp!=NULL)
                    {
                        if(tmp->next[i]!=NULL)
                        {
                            p->next[i]->fail=tmp->next[i];
                            break;
                        }
                        tmp=tmp->fail;
                    }
                    if(tmp==NULL)
                        p->next[i]->fail=root;
                }
                q[tail++]=p->next[i];
            }
        }
    }
}
int query(char *s)
{
    int i,j,len=strlen(s),flag=0;
    node *p=root,*tmp;
    memset(vis,0,sizeof(vis));
    for(i=0;i<len;i++)
    {
        j=s[i]-32;
        while(p->next[j]==NULL&&p!=root)
            p=p->fail;
            p=p->next[j];
        if(p==NULL)
            p=root;
        tmp=p;
        while(tmp!=NULL)
        {
            if(tmp->count>0)
            {
                flag=1;
                vis[tmp->id]=1;
            }
            tmp=tmp->fail;
        }
    }
    return flag;
}
int main()
{
    int n,m,i,j;
    while(scanf("%d",&n)!=EOF)
    {
        root=new node();
        for(i=0;i<n;i++)
        {
            scanf("%s",str);
            insert(str,i);
        }
        build();
        scanf("%d",&m);
        int ans=0;
        for(i=0;i<m;i++)
        {
            scanf("%s",s);
            if(query(s))
            {
                ans++;
                printf("web %d:",i+1);
                for(j=1;j<=n;j++)
                    if(vis[j])
                    printf(" %d",j);
                    printf("\n");
            }
        }
        printf("total: %d\n",ans);
    }
    return 0;
}