HDU 2896 病毒侵袭(AC自动机)

分析:这也是一道很裸的AC自动机,单词结尾的节点不在是1,而是id,即这个单词的序号,然后用vector把所有序号存在来,一遍去重,然后输出答案就好了。

总的复杂度为o(m*l)

代码:

#include <bits/stdc++.h>
#define LL long long
#define uLL long long
#define FOR(i,x,y)  for(int i = x;i < y;++ i)
#define IFOR(i,x,y) for(int i = x;i > y;-- i)

using namespace std;

typedef vector <int> VT;

const int maxn = 550*220;
const int maxm = 10010;

struct Trie{
    int ch[maxn][128],en[maxn],fail[maxn];
    int rt,sz;

    void init(){
        rt = sz = 0;
        memset(ch[rt],-1,sizeof(ch[rt]));
    }

    void newnode(){
        ++ sz;
        memset(ch[sz],-1,sizeof(ch[sz]));
        en[sz] = 0;
    }

    void insert(char* str,int val){
        int len = strlen(str);
        int u = rt;
        FOR(i,0,len){
            if(ch[u][str[i]-' '] == -1){
                newnode();
                ch[u][str[i]-' '] = sz;
            }
            u = ch[u][str[i]-' '];
        }
        en[sz] = val;
    }

    void build(){
        queue <int> q;
        FOR(i,0,128){
            if(ch[rt][i] == -1){
                ch[rt][i] = rt;
            }
            else{
                fail[ch[rt][i]] = rt;
                q.push(ch[rt][i]);
            }
        }
        while(!q.empty()){
            int u = q.front();  q.pop();
            FOR(i,0,128){
                if(ch[u][i] == -1){
                    ch[u][i] = ch[fail[u]][i];
                }
                else{
                    fail[ch[u][i]] = ch[fail[u]][i];
                    q.push(ch[u][i]);
                }
            }
        }
    }

    VT query(char* str){
        int len = strlen(str);
        int now = rt;
        VT res;
        FOR(i,0,len){
            now = ch[now][str[i]-' '];
            int temp = now;
            while(temp != rt){
                if(en[temp]){
                    res.push_back(en[temp]);
                }
                temp = fail[temp];
            }
        }
        return res;
    }
}ac;

int n,m;
char s[maxm];

int main()
{
    //freopen("test.in","r",stdin);
    while(~scanf("%d",&n)){
        ac.init();
        FOR(i,1,n+1){
            scanf("%s",s);
            ac.insert(s,i);
        }
        ac.build();
        scanf("%d",&m);
        int id = 0;
        FOR(i,1,m+1){
            scanf("%s",s);
            VT ans = ac.query(s);
            if(ans.size()){
                ++id;
                printf("web %d:",i);
                sort(ans.begin(),ans.end());
                VT :: iterator it = unique(ans.begin(),ans.end());
                ans.erase(it,ans.end());
                FOR(i,0,(int)ans.size()){
                    printf(" %d",ans[i]);
                }
                printf("\n");
            }
        }
        printf("total: %d\n",id);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值