洛谷P3808 【模板】AC自动机(简单版)

#include <bits/stdc++.h>
#define N 500010
using namespace std;
// typedef long long ll;
inline int read() {
    int x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9')
        x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
    return x * f;
}

queue<int> q;
struct Aho_Corasick_Automaton {
    int c[N][26], val[N], fail[N], cnt;
    void ins(char* s) {  //初始化建树
        int len = strlen(s);
        int now = 0;
        for (int i = 0; i < len; i++) {
            int v = s[i] - 'a';
            if (!c[now][v])
                c[now][v] = ++cnt;
            now = c[now][v];
        }
        val[now]++;  //以这个节点结尾的字符串有几个
    }
    void build() {  //建立fail数组
        for (int i = 0; i < 26; i++)
            if (c[0][i]) {          //如果结点存在
                fail[c[0][i]] = 0;  // 0就是指向根结点
                q.push(c[0][i]);    //把结点标号放入队列
            }
        while (!q.empty()) {
            int u = q.front();
            q.pop();
            for (int i = 0; i < 26; i++)  //遍历结点的下一层
                if (c[u][i]) {
                    fail[c[u][i]] = c[fail[u]][i];
                    q.push(c[u][i]);
                    //如果有这个子节点为字母i+'a',则
                    //让这个节点的失败指针指向(((他父亲节点)的失败指针所指向的那个节点)的下一个节点)
                } else
                    //否则就让当前节点的这个子节点
                    //指向当前节点fail指针的这个子节点
                    c[u][i] = c[fail[u]][i];
            
        } 
    }
    int query(char* s) {  //查询
        int len = strlen(s);
        int now = 0, ans = 0;
        for (int i = 0; i < len; i++) {
            now = c[now][s[i] - 'a'];
            for (int t = now; t && ~val[t]; t = fail[t])
                ans += val[t], val[t] = -1;
        }
        return ans;
    }
} AC;
int n;
char p[1000005];
int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%s", p), AC.ins(p);
    AC.build();
    scanf("%s", p);
    int ans = AC.query(p);
    printf("%d\n", ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值