hdu 2896 病毒侵袭 -- AC自动机

/*

寻找都有哪些子串

不能保证是字母或数字，所以子节点有差不多130个

*/

#include <iostream> 

#include<string.h>

#include<stdlib.h>

using namespace std; 

  

int biaoshi[510];

const int kind = 130; 

int dulist[10];

struct node{  

    node *fail;       //失败指针

    node *next[kind]; 

    int count;        //是否为该单词的最后一个节点

    node(){           //构造函数初始化

        fail=NULL; 

        count=0; 

        memset(next,NULL,sizeof(next)); 

    } 

}*q[5000000];          //队列，方便用于bfs构造失败指针

char keyword[510];     //输入的单词

char str[1000010];    //模式串

int head,tail;        //队列的头尾指针

  

void insert(char *str,node *root,int no){ 

    node *p=root; 

    int i=0,index;  

    while(str[i]){ 

        index=str[i]-31; 

        if(p->next[index]==NULL) p->next[index]=new node();  

        p=p->next[index];

        i++;

    } 

    p->count=no;//初始化为0，++后为1，表示是一个单词的结尾

} 

void build_ac_automation(node *root)

{

    int i;

    root->fail=NULL; 

    q[head++]=root; 

    while(head!=tail)

	{ 

        node *temp=q[tail++];//拿到一个节点

        node *p=NULL; 

        for(i=0;i<130;i++)

		{ 

            if(temp->next[i]!=NULL)//若其i孩子非空

			{ 

                if(temp==root) //他自己是头，其孩子的失败指针指向头

					temp->next[i]->fail=root;                 

                else{ //普通节点

                    p=temp->fail; //指向自己的失败指针

                    while(p!=NULL)

					{  

                        if(p->next[i]!=NULL)//失败指针有i孩子

						{ 

                            temp->next[i]->fail=p->next[i]; //当前节点的i孩子的失败指针指向失败指针的i孩子，然后跳出

                            break; 

                        } 

                        p=p->fail; //继续找失败指针

                    } 

                    if(p==NULL) //若失败指针为空

						temp->next[i]->fail=root; //当前节点的i孩子的失败指针指向头

                } 

                q[head++]=temp->next[i];  //进去的都是定义过失败指针的,故此过程是给其孩子定义失败指针

            } 

        }   

    } 

} 

int query(node *root)

{ 

    int i=0,cnt=0,index,len=strlen(str); 

    node *p=root;  

    while(str[i])

	{  

        index=str[i]-31;  //计算孩子的位置

        while(p->next[index]==NULL && p!=root) 

			p=p->fail; //若没有i孩子节点，通过失败指针找与这之前相同的有i孩子的节点

        p=p->next[index]; //指向其i孩子

        p=(p==NULL)?root:p; 

        node *temp=p; 

        while(temp!=root && temp->count)

		{ 

			if(temp->count&&!biaoshi[temp->count])//主要是改这里

			{

				cnt++;

				biaoshi[temp->count]=1;

			}

            temp=temp->fail; 

        } 

        i++;                 

    }    

    return cnt; 

} 

int cmp(const void *a,const void *b)

{

	int *c=(int*)a,*d=(int*)b;

	return *c-*d;

}

void print(int ji,int na,int n)

{



	printf("web %d:",na);

	for(int i=1;i<=n;++i)

	{

		if(biaoshi[i])

		{

		printf(" %d",i);

		}

	}

	printf("\n");

}

int main(){ 

    int n,t,i,j=0; 

    while(scanf("%d",&n)!=EOF)

	{  

        head=tail=0; 

        node *root=new node(); 

      

        getchar(); 

        for(i=1;i<=n;i++)

		{

            gets(keyword); 

            insert(keyword,root,i); 

        } 

        build_ac_automation(root); 

		scanf("%d",&t);

		for(i=1;i<=t;++i)

		{

			memset(biaoshi,0,sizeof(biaoshi));

			scanf("%s",str); 

			int ge=query(root);

			if(ge)

			{

				print(ge,i,n);

				j++;

			}

		}

		printf("total: %d\n",j);

    } 

    return 0; 

}