HDU2896 病毒侵袭(AC自动机)

传送门
这一题实际上就是多模式,多文本的匹配,对于这道题来说,用AC自动机就可以解决,题解同前一道题。
代码:

#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 10005
#define MAXC 128
#define MAXM 505
#define MAXL 205
int n, root, sz, q[MAXM*MAXL];
vector<int> arr;
char s[MAXN];
struct Trie {
    int nxt[MAXC], fail, id;
} t[MAXM * MAXL];
inline int NewNode() {
    return ++ sz;
}
inline void Ins(char *s, int id) {
    int len = strlen(s), cur = root;
    for(int i = 0; i < len; ++ i) {
        if(!t[cur].nxt[s[i]]) t[cur].nxt[s[i]] = NewNode();
        cur = t[cur].nxt[s[i]];
    }
    t[cur].id = id;
}
inline void Build() {
    int l = 1, r = 0; t[root].id = -1;
    for(int i = 0; i < MAXC; ++ i)
        if(!t[root].nxt[i]) t[root].nxt[i] = root;
        else {
            t[ t[root].nxt[i] ].fail = root;
            q[++ r] = t[root].nxt[i];
        }
    while(l <= r) {
        int u = q[l ++];
        for(int i = 0; i < MAXC; ++ i)
            if(!t[u].nxt[i]) t[u].nxt[i] = t[ t[u].fail ].nxt[i];
            else {
                t[ t[u].nxt[i] ].fail = t[ t[u].fail ].nxt[i];
                q[++ r] = t[u].nxt[i];
            }
    }
}
inline int Query(char *s) {
    int len = strlen(s), cur = root, ans = 0;
    for(int i = 0; i < len; ++ i) {
        cur = t[cur].nxt[s[i]];
        int tmp = cur;
        while(~t[tmp].id) {
            if(t[tmp].id) arr.push_back(t[tmp].id);
            tmp = t[tmp].fail;
        }
    }
    return ans;
}
int main() {
    sz = 0; root = NewNode();
    scanf("%d", &n);
    for(int i = 1; i <= n; ++ i) {
        scanf("%s", s);
        Ins(s, i);
    }
    Build(); int Count = 0;
    scanf("%d", &n);
    for(int i = 1; i <= n; ++ i) {
        scanf("%s", s);
        Query(s);
        if(arr.size()) {
            ++ Count;
            sort(arr.begin(), arr.end());
            printf("web %d:", i);
            for(int j = 0; j < (int)arr.size(); ++ j)
                printf(" %d", arr[j]);
            putchar('\n');
            arr.clear();
        }
    }
    printf("total: %d\n", Count);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值