C++怎么实现一个后缀自动机(SAM)_C++高级字符串算法与子串问题

后缀自动机能在线性时间内构建并解决子串相关问题,其核心是通过状态转移和后缀链接表示所有子串。

c++怎么实现一个后缀自动机(sam)_c++高级字符串算法与子串问题

后缀自动机(Suffix Automaton,简称 SAM)是处理字符串子串问题的高效数据结构。它能在 O(n) 时间内构建,并支持快速解决子串计数、最长公共子串、不同子串个数等问题。下面介绍 C++ 中如何实现一个基础的后缀自动机。

什么是后缀自动机

后缀自动机是一个有向图,其中每个节点代表原字符串某个子串的“状态”,边表示在该状态后添加一个字符能转移到的新状态。SAM 的核心性质包括:

所有从初始状态可达的路径,对应原串的所有子串。总状态数不超过 2n,转移边数不超过 3n。每个状态记录了该状态所代表子串的最长长度(len)和最小长度(通过 link 树推导)。

核心结构与变量定义

我们用结构体维护 SAM 的每个节点,主要包含以下字段:

len:当前状态能表示的最长子串长度。link:后缀链接(suffix link),指向另一个状态,表示当前状态的最长串的真后缀中最长的可接受状态。next[26]:转移数组,记录添加字符 a~z 后的下一个状态。

// 假设只处理小写字母struct State {int len; // 最长子串长度int link; // 后缀链接int next[26]; // 转移边State() {len = 0;link = -1;for (int i = 0; i

全局维护一个状态数组和相关变量:

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

vector st;int last = 0; // 当前最后一个字符所在的状态int sz = 0; // 状态总数

构建后缀自动机

每次添加一个字符时,创建新状态并更新后缀链接。算法核心步骤如下:

void sam_init() {st.push_back(State());sz++;last = 0;}

void sam_extend(char c) {int cur = sz++;st.push_back(State());st[cur].len = st[last].len + 1;int p = last;

// 从 last 开始,沿着后缀链接走,直到根或已有 c 转移while (p != -1 && st[p].next[c - 'a'] == -1) {    st[p].next[c - 'a'] = cur;    p = st[p].link;}if (p == -1) {    st[cur].link = 0;} else {    int q = st[p].next[c - 'a'];    if (st[p].len + 1 == st[q].len) {        st[cur].link = q;    } else {        int clone = sz++;        st.push_back(st[q]);  // 复制 q 的信息        st[clone].len = st[p].len + 1;        // 将 q 的转移复制,但不改变 len 和 link        while (p != -1 && st[p].next[c - 'a'] == q) {            st[p].next[c - 'a'] = clone;            p = st[p].link;        }        st[q].link = st[cur].link = clone;    }}last = cur;

}

说明:

每次扩展字符 c,新建状态 cur,其最长长度为 last.len + 1。从 last 沿着 link 往上跳,为没有 c 转移的状态添加指向 cur 的边。若遇到已有 c 转移的状态 p,则检查目标状态 q 是否满足 len 条件。若不满足(即 q 是被多个路径共享的复杂状态),则需分裂出一个 clone 状态来保证正确性。

常见应用示例

利用 SAM 可以高效求解多种问题:

1. 不同子串个数

每个状态表示若干长度连续的子串,数量为 len – len(link)。累加所有状态即可。

long long count_distinct_substrings() { long long total = 0; for (int i = 1; i

2. 最短未出现子串

从初始状态开始 BFS,找第一个缺失的字符转移,该字符即是最短未出现子串。

3. 最长公共子串(两个串)

对第一个串建 SAM,然后用第二个串在 SAM 上匹配,维护当前匹配长度和状态,不断沿 link 跳转以缩短匹配。

int longest_common_substring(string s, string t) {int res = 0, v = 0, l = 0;for (char c : t) {while (v && st[v].next[c-‘a’] == -1) {v = st[v].link;l = st[v].len;}if (st[v].next[c-‘a’] != -1) {v = st[v].next[c-‘a’];l++;}res = max(res, l);}return res;}

基本上就这些。SAM 虽初看复杂,但掌握 extend 过程和 link 作用后,理解会清晰很多。关键是理解 clone 操作的意义:保持 DAG 中每个状态的 right 集合一致。

以上就是C++怎么实现一个后缀自动机(SAM)_C++高级字符串算法与子串问题的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1488541.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月19日 11:38:26
下一篇 2025年12月19日 11:38:37

相关推荐

发表回复

登录后才能评论
关注微信