C++怎么实现一个AC自动机_C++高效解决多模式串匹配问题的算法

2次阅读

AC自动机通过构建Trie树、失败指针和执行匹配实现高效多模式串查找,时间复杂度为O(n + m + z),适用于大规模关键词匹配。

C++怎么实现一个AC自动机_C++高效解决多模式串匹配问题的算法

AC自动机(Aho-Corasick Automaton)是一种用于高效解决多模式串匹配的经典算法。当需要在一个主串中查找多个模式串是否出现时,使用AC自动机比逐个进行KMP或暴力匹配要高效得多。c++实现AC自动机主要包含三个步骤:构建Trie树、构建失败指针(fail指针)、执行匹配。

构建Trie树

将所有模式串插入到Trie树中,每个节点代表一个字符,路径从根到叶表示一个完整的模式串。同时在每个节点记录是否为某个模式串的结尾,并保存对应的模式串编号或出现次数。

– 每个节点用数组或map存储子节点指针 – 设置一个标记变量表示该节点是否为某个模式串的结束 – 可额外记录模式串索引或数量

示例结构:

struct node {     int next[26]; // 假设只有小写字母     bool isEnd;     int id;       // 模式串编号     Node() {         fill(next, next + 26, -1);         isEnd = false;         id = -1;     } }; vector<Node> trie(1); // 初始化根节点

构建失败指针(Fail指针)

失败指针的作用类似于KMP中的next数组,用于在匹配失败时跳转到最长公共前后缀的位置。通过BFS遍历Trie树来构建fail指针。

立即学习C++免费学习笔记(深入)”;

– 根节点的所有直接子节点的fail指向根 – 对于当前节点u的子节点v,查找trie[u].fail对应节点是否有相同字符的子节点 – 如果有,则v的fail指向那个子节点;否则继续沿fail链向上找 – 若最终没找到,指向根节点

BFS过程伪代码逻辑:

C++怎么实现一个AC自动机_C++高效解决多模式串匹配问题的算法

拾贝

一键同步微信读书所有笔记和划线,并在新标签页回顾

C++怎么实现一个AC自动机_C++高效解决多模式串匹配问题的算法 186

查看详情 C++怎么实现一个AC自动机_C++高效解决多模式串匹配问题的算法

queue<int> q; for (int i = 0; i < 26; ++i) {     if (trie[0].next[i] != -1) {         int child = trie[0].next[i];         fail[child] = 0;         q.push(child);     } } while (!q.empty()) {     int u = q.front(); q.pop();     for (int i = 0; i < 26; ++i) {         int &v = trie[u].next[i];         int f = fail[u];         if (v != -1) {             while (f != -1 && trie[f].next[i] == -1) f = fail[f];             fail[v] = (f == -1) ? 0 : trie[f].next[i];             q.push(v);         }     } }

执行多模式匹配

从主串第一个字符开始,在Trie树上逐字符转移状态。如果当前节点没有对应子节点,则通过fail指针回溯,直到可以转移或回到根节点。

– 遍历主串每个字符c – 当前状态为cur,尝试转移到trie[cur].next[c-‘a’] – 若无法转移,通过fail链寻找可转移位置 – 每到达一个节点,沿fail链回溯所有可能的模式串结尾并记录结果

关键匹配逻辑:

int cur = 0; for (char c : text) {     int idx = c - 'a';     while (cur != -1 && trie[cur].next[idx] == -1)         cur = fail[cur];     cur = (cur == -1) ? 0 : trie[cur].next[idx]; <pre class="brush:php;toolbar:false;">int temp = cur; while (temp != 0) {     if (trie[temp].isEnd) {         cout << "Pattern found at position: "              << i - pattern_len + 1 << endl;     }     temp = fail[temp]; }

}

优化建议:

  • 使用静态数组代替vector以提升性能
  • 合并重复模式串避免冗余
  • 在构建fail指针时同时更新输出链(output link),避免每次匹配都遍历fail链

基本上就这些。AC自动机的时间复杂度为O(n + m + z),其中n是主串长度,m是所有模式串总长度,z是匹配次数,非常适合大规模多关键词匹配场景。

text=ZqhQzanResources