kuangbin專題十七 HDU2896 病毒侵襲(AC自動機模板題)

題意:
中文題。
題解:
這道題有點狗血的就是它的特徵碼不單單是小寫字母。。。所以要寫成128,囊括所有的ASCII碼才行。。
因爲它要你計算多個網站分別中了幾個病毒,那麼我們可以在模板上加個vis數組表示的是該病毒是否已經出現過了,出現過的話就不用再加上去了。

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
const int MAXN=10000+7;
struct node
{
    int sum;
    int number;
    node *fail;
    node *next[128];//坑爹玩意,特徵碼不單單是小寫字母。。128位ASCII碼要全部囊括起來。 
    node(){
        number=0;
        sum=0;
        fail=0;
        for(int i=0;i<128;i++)
        next[i]=0;
    }
}*root;
char str[MAXN];
int tot;
int cnt;
int ans[507],k;
bool vis[507];
void insert(char *s)
{
    node *r=root;
    for(int i=0;s[i];i++)
    {
        int x=s[i];
        if(r->next[x]==0) r->next[x]=new(node);
        r=r->next[x]; 
    }
    r->number=tot++;
    r->sum++;
}
void getfail()
{
    queue<node *>q;
    q.push(root);
    node *p;
    node *temp;
    while(!q.empty())
    {
        temp=q.front();
        q.pop();
        for(int i=0;i<128;i++)
        {
            if(temp->next[i])
            {
                if(temp==root)
                {
                    temp->next[i]->fail=root;
                }
                else
                {
                    p=temp->fail;
                    while(p)
                    {
                        if(p->next[i])
                        {
                            temp->next[i]->fail=p->next[i];
                            break;
                        }
                        p=p->fail;
                    }
                    if(p==0)
                    temp->next[i]->fail=root;
                }
                q.push(temp->next[i]);
            }
        }
    }
}
void ac_automation(char *s)
{
    node *p=root;
    int len=strlen(s);
    for(int i=0;i<len;i++)
    {
        int x=s[i];
        while(!p->next[x]&&p!=root) p=p->fail;
        p=p->next[x];
        if(!p) p=root;
        node *temp=p;
        while(temp!=root)
        {
            if(temp->sum>0&&!vis[temp->number])
            {
                ans[k++]=temp->number;
                vis[temp->number]=true;
            }
            temp=temp->fail;
        }
    }
}
void del(node *head)
{
    for(int i=0;i<128;i++)
    if(head->next[i])
    del(head->next[i]);
    delete(head);
}
int main()
{
    int n,m;
    while(~scanf("%d",&n))
    {
        root=new(node);
        tot=1;
        cnt=0;
        char c[207];
        for(int i=0;i<n;i++)
        {
            scanf("%s",c);
            insert(c);
        }
        getfail();
        scanf("%d",&m); 
        for(int i=0;i<m;i++)
        {
            scanf("%s",str);
            memset(vis,false,sizeof(vis));
            k=0;
            ac_automation(str);
            sort(ans,ans+k);
            if(k)
            printf("web %d: ",i+1),cnt++;
            for(int j=0;j<k;j++)
            if(j!=k-1)
            printf("%d ",ans[j]);
            else
            printf("%d\n",ans[j]);
        }
        printf("total: %d\n",cnt);
        del(root);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章