hdu 2896 病毒侵袭

http://acm.hdu.edu.cn/showproblem.php?pid=2896

思路：建立字典树时，把字符串单词节点信息改为id就可以，然后计算失配函数，再查找哪一个上面的病毒字符串在下面的网站信息中出现，AC自动机。

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;

int n,m;
char str[2000000];
set<int>g;
struct node
{
    struct node *next[128],*fail;
    int v;
}*root;

void insert(char str[],int v)
{
    struct node *head=root;
    struct node *temp;
    int k=strlen(str);
    for(int i=0; i<k; i++)
    {
        int x=(int)str[i];
        if(head->next[x]==NULL)
        {
            temp=(struct node *)malloc(sizeof(struct node));
            temp->v=0;
            temp->fail=root;
            for(int j=0; j<128; j++)
            {
                temp->next[j]=NULL;
            }
            head->next[x]=temp;
        }
        head=head->next[x];
        if(i==k-1) head->v=v;
    }
}

void Build_AC()
{
    struct node *head=root;
    queue<node *>q;
    q.push(root);
    while(!q.empty())
    {
        head=q.front();
        q.pop();
        for(int j=0; j<128; j++)
        {
            if(head->next[j])
            {
                q.push(head->next[j]);
                if(head==root)
                {
                    head->next[j]->fail=root;
                }
                else
                {
                    struct node *p=head->fail;
                    bool flag=false;
                    while(p->next[j]==NULL)
                    {
                        if(p==root)
                        {
                            flag=true;
                            break;
                        }
                        p=p->fail;
                    }
                    if(flag)
                    {
                        head->next[j]->fail=root;
                    }
                    else
                    {
                        head->next[j]->fail=p->next[j];
                    }
                }
            }
        }
    }
}

void Find(char str[])
{
    struct node *head=root,*p;
    int k=strlen(str);
    for(int i=0; i<k; i++)
    {
        int x=(int)str[i];
        while(head->next[x]==NULL&&head!=root) head=head->fail;
        if(head->next[x]==NULL) head=root;
        else head=head->next[x];
        p=head;
        while(p!=root)
        {
            if(p->v)
            g.insert(p->v);
            p=p->fail;
        }
    }
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        root=(struct node*)malloc(sizeof(struct node));
        root->fail=root;
        root->v=0;
        for(int i=0; i<128; i++)
        {
            root->next[i]=NULL;
        }
        for(int i=1; i<=n; i++)
        {
            scanf("%s",str);
            insert(str,i);
        }
        Build_AC();
        set<int>::iterator it;
        int ans=0;
        scanf("%d",&m);
        for(int j=1; j<=m; j++)
        {
            g.clear();
            scanf("%s",str);
            Find(str);
            if(g.size())
            {
                ans++;
                printf("web %d:",j);
                for(it=g.begin(); it!=g.end(); it++)
                {
                    printf(" %d",*it);
                }
                printf("
");
            }
        }
        printf("total: %d
",ans);
    }
    return 0;
}