HDU2896(病毒侵袭--AC自动机)

来源：互联网发布：昆工教务网络管理系统编辑：程序博客网时间：2024/05/05 06:03
题目：病毒侵袭
#include <iostream>#include <string.h>#include <stdio.h>#include <algorithm>#include <queue>using namespace std;const int N=1000050;char S[N];char keyword[256];class Trie{    public:        int index;        Trie *fail;        Trie *next[128];        Trie()        {            index=0;            fail=NULL;            memset(next,NULL,sizeof(next));        }};Trie *root;queue<Trie*>Q;int ss[5500][5],k[5500];void Insert(char *S,int value){    int len=strlen(S);    Trie *p=root;    for(int i=0; i<len; i++)    {        int id=S[i]-' ';        if(p->next[id]==NULL)            p->next[id]=new Trie();        p=p->next[id];    }    p->index=value;}void Build_AC(){    Q.push(root);    root->fail=NULL;    while(!Q.empty())    {        Trie *p=NULL;        Trie *tmp=Q.front();        Q.pop();        for(int i=0; i<128; i++)        {            if(tmp->next[i])            {                if(tmp==root) tmp->next[i]->fail=root;                else                {                    p=tmp->fail;                    while(p)                    {                        if(p->next[i])                        {                            tmp->next[i]->fail=p->next[i];                            break;                        }                        p=p->fail;                    }                    if(p==NULL) tmp->next[i]->fail=root;                }                Q.push(tmp->next[i]);            }        }    }}void Query(char *S,int c){    Trie *p=root;    int flag[550];    k[c]=0;    memset(flag,0,sizeof(flag));    int index;    int len=strlen(S);    for(int i=0; i<len; i++)    {        index=S[i]-' ';        while(p->next[index]==NULL&&p!=root) p=p->fail;        p=p->next[index];        if(p==NULL) p=root;        Trie *tmp=p;        while(tmp!=root&&!flag[tmp->index])        {            if(tmp->index)            {                ss[c][k[c]++]=tmp->index;                flag[tmp->index]=1;            }            tmp=tmp->fail;        }        if(k[c]>=3) break;    }}int main(){    int t,n,m;    while(~scanf("%d",&n))    {        root=new Trie();        getchar();        for(int i=1; i<=n; i++)        {            gets(keyword);            Insert(keyword,i);        }        Build_AC();        scanf("%d",&m);        int total=0;        for(int i=1; i<=m; i++)        {            scanf("%s",S);            Query(S,i);            if(k[i])            {                sort(ss[i],ss[i]+k[i]);                total++;            }        }        for(int i=1; i<=m; i++)        {            if(!k[i]) continue;            printf("web %d:",i);            for(int j=0; j<k[i]; j++)                printf(" %d",ss[i][j]);            puts("");        }        printf("total: %d\n",total);    }    return 0;}