AC自动机模板

终于写出来了传说中的AC自动机,听起来这个名字就非常的吉利,AC啊……此版本AC自动机是照着佳哥的《训练指南》写的,实现的功能是查找主串种出现了多少种模式串。AC自动机简单来说就是在字典树上指失配指针,然后进行匹配,当然,我这个AC自动机是有了Trie图的,效率能更高一些。

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
int ans = 0;
struct ACautomata{
    int ch[500001][27], val[500001];
    int f[500010], sz;
    void reset(){
        memset(ch[0], 0, sizeof(ch[0]));
        memset(val, 0, sizeof(val));
        sz = 1;
    }
    int idx(char c){
        return c - 'a';
    }
    void insert(char *s, int v){
        int u = 0, n = strlen(s);
        for (int i = 0; i < n; i++){
            int c = idx(s[i]);
            if (!ch[u][c]){
                memset(ch[sz], 0, sizeof(ch[sz]));
                val[sz] = 0;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }
    void getfail(){
        queue<int> q;
        f[0] = 0;
        for (int c = 0; c < 26; c++){
            int u = ch[0][c];
            if (u){f[u] = 0; q.push(u);}
        }
        while(!q.empty()){
            int r = q.front(); q.pop();
            for (int c = 0; c < 26; c++){
                int u = ch[r][c];
                if (!u){ch[r][c] = ch[f[r]][c]; continue;}
                q.push(u);
                int v = f[r];
                while(v && !ch[v][c]) v = f[v];
                f[u] = ch[v][c];
            }
        }
    }
    void find(char *T){
        int n = strlen(T);
        int j = 0;
        for (int i = 0; i < n; i++){
            int c = idx(T[i]);
            j = ch[j][c];
            int tmp = j;
            while(tmp && val[tmp]){
                ans += 1;
                val[tmp] = 0;
                tmp = f[tmp];
            }
        }
    }
}ac;
char s[55], ss[11111];
int main(){
    int t;
    ac.reset();
    scanf("%d", &t);
    while(t--){
        int n;
        scanf("%d", &n);
        while (n--){
            scanf("%s", s);
            ac.insert(s, 1);
        }
        ac.getfail();
        scanf("%s", ss);
        ans = 0;
        ac.find(ss);
        printf("%d\n", ans);
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值