首页 > 代码库 > AC自动机 病毒侵袭 hdu2896

AC自动机 病毒侵袭 hdu2896

和hdu2222题相似的水题

提示:

1)连着RE了好多发,没想明白,看了一下网上题解才知道,输入的不一定都是字母,所以next要开100!!!!!!!

#include <stdio.h>
#include <string.h>
int tot;
char str[10005];
int t;
//int time[100];
struct trie {
    trie *fail;
    trie *next[100];
    int num;
    trie() {
        for(int i=0; i<100; i++)
            next[i]=NULL;
        fail=NULL;
        num=0;
    }
}*queue[100005];
struct xxx
{
    int virus[505];
    int vt;
}web[1005];
trie *root;
void build() {                 //建立字典树
    int len=strlen(str);
    trie *p=root;
    for(int i=0;i<len;i++) {
        int id=str[i]-' ';
        if(p->next[id]==NULL)
            p->next[id]=new trie();
        p=p->next[id];
    }
    if(p->num==0)
        p->num=tot++;         //给每一个单词结尾附上序号
}
void build_fail() {           //构建失败指针
    int i;
    trie *u,*v;
    int fr,ed;
    fr=ed=0;
    root->fail=NULL;
    queue[ed++]=root;
    while(fr<ed) {
        u=queue[fr++];
        v=NULL;
        for(i=0; i<100; i++) {
            if(u->next[i]!=NULL) {
                if(u==root)
                    u->next[i]->fail=root;
                else {
                    v=u->fail;
                    while(v!=NULL) {
                        if(v->next[i]!=NULL) {
                            u->next[i]->fail=v->next[i];
                            break;
                        }
                        v=v->fail;
                    }
                    if(v==NULL)
                        u->next[i]->fail=root;
                }
                queue[ed++]=u->next[i];
            }
        }
    }
}
void search(int n) {            //查找字符串
    int i=0;
    trie *p=root;
    while(str[i]) {
        int id=str[i]-' ';
        while(p->next[id]==NULL&&p!=root)
            p=p->fail;
        p=p->next[id];
        p=(p==NULL)?root:p;
        trie *q=p;
        while(q!=root)
        {
            if(q->num!=0)
            {
                web[n].virus[q->num]=1;    //记录病毒种类
                web[n].vt++;          //统计个数
            }
            q=q->fail;
        }
        i++;
    }
}
int main() {
    int n;
    int i;
    scanf("%d",&n);
    tot=1;
    memset(web,0,sizeof(web));
    root=new trie();
    for(i=0; i<n; i++) {
        scanf("%s",str);
        build();
    }
    build_fail();
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",str);
        search(i);
    }
    t=0;
    for(int i=1;i<=n;i++)
    {
        if(web[i].vt!=0)
        {
            t++;
            printf("web %d:",i);
            for(int j=0;j<505;j++)
                if(web[i].virus[j]!=0)
                    printf(" %d",j);
            printf("\n");
        }
    }
    printf("total: %d\n",t);
    return 0;
}


AC自动机 病毒侵袭 hdu2896