【AC自动机】HDU2896病毒侵袭

来源：互联网发布：python 余弦相似度包编辑：程序博客网时间：2024/04/25 23:46

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2896

这个在模板的基础之上增加一个标记就可以了，保存每一个出现的病毒就可以了；所以啊，代码里的cnt是一个变化很灵活的变量；

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<queue>using namespace std;const int N=130;        //  ASCII码可见字符0~127;char str[10005];int vis[555];struct node{    int cnt;    node *fail;    node *next[N];    node(){        cnt=0;        fail=NULL;        memset(next,0,sizeof(next));    }};node *p,*temp,*head;void Insert(char s[],int num)       //  第一步，构造字典树；{    p=head;    int i=0,id;    while(s[i]){        id=s[i++]-' ';        if(p->next[id]==NULL) p->next[id]=new node();        p=p->next[id];    }    p->cnt=num;             //  保存这是第几个病毒；}void get_fail()             //  构造fail指针；{    node *son;    p=head;    queue<struct node *>q;    q.push(p);    while(!q.empty()){          //  广搜，构造所有的fail指针；        temp=q.front();        q.pop();        for(int i=0;i<128;i++){                 //  遍历所有可能的子节点；            son=temp->next[i];            if(son!=NULL){                      //  子节点不为空时；                if(temp==head) son->fail=head;  //  当子节点为第二层节点时；                else{                    p=temp->fail;               //  当不为前两层节点时；                    while(p){                        if(p->next[i]){         //  如果父节点的fail指针的下一层存在，那么改子节点的fail便是指向那里；                            son->fail=p->next[i];                            break;              //  构造成功，便可以直接退出；                        }                        p=p->fail;                    }                    if(!p) son->fail=head;      //  如果没有构造成功，那么fail便只有指向head了；                }                q.push(son);                    //  新元素入队；            }        }    }}int Query()                     //  查询；{    int i=0,ans=0;    p=head;    while(str[i]){        int id=str[i++]-' ';        while(p->next[id]==NULL&&p!=head) p=p->fail;    //  这个字符的下一个不存在，且不是第一层字符节点；        p=p->next[id];          //  将p更新到该有的位置；        if(!p) p=head;          //  如果不存在；        temp=p;        while(temp!=head){            vis[temp->cnt]=1;               //  标记每一个出现的病毒；            if(temp->cnt!=0) ans++;            temp=temp->fail;        }    }    return ans;}int main(){    int n,m;    head=new node();    scanf("%d",&n);    int ans=0;    for(int i=1;i<=n;i++){          //  构造字典树；        scanf("%s",str);        Insert(str,i);    }    get_fail();                     //  构造fail指针；    scanf("%d",&m);    for(int i=1;i<=m;i++){        scanf("%s",str);        memset(vis,0,sizeof(vis));        int tmp=Query();        if(tmp){            ans++;            printf("web %d:",i);            for(int j=1;j<=n;j++)                if(vis[j]) printf(" %d",j);            printf("\n");        }    }    printf("total: %d\n",ans);    return 0;}

0 0