高级数据结构-09:KMP 算法
基本术语
前缀与后缀
字符串是线性连续的数据序列。为避免索引歧义,通常用半开半闭区间 表示字符串片段,其中包含起始索引 ,不包含终止索引 。
设字符串 长度为 ,有效索引区间为 。
字符串有两类特殊子串:前缀与后缀。前缀 表示从起始位置开始、长度为 的片段,等价于 或 。后缀表示最后 个字符组成的子串:

任意内部子串都可以表示为某个前缀的后缀,或某个后缀的前缀:。这种区间恒等关系为后续复用已知匹配信息提供了基础。
抽象数据类型接口
字符串抽象数据类型(ADT)提供标准操作接口,支持属性获取、子串截取、拼接和模式搜索等操作。

应用实例
以主字符串 "data structures" 为例,length() 返回 。在零起点索引下,charAt(5) 返回字符 's'。prefix(4) 截取首部片段 "data";suffix(10) 从尾部截取,得到 "structures"。
在涉及多个字符串的运算时,"data structures".concat(" & algorithms") 会在内存中申请新的空间,拼接生成 "data structures & algorithms"。若对不同字符串进行严格一致性判定,"algorithms".equal("data structures") 将返回逻辑假 false。
对于核心的模式检索,若在文本 "data structures and algorithms" 中调用 indexOf("string"),由于该子序列未在主序列中出现,函数将返回状态码 -1 表示未命中。若调用 indexOf("algorithm"),检索过程将在主串索引 处完成匹配,并返回该整型位置数据。
字符串匹配
循模式访问
模式匹配算法的性能依赖输入分布。设文本串 长度为 ,模式串 长度为 。典型场景通常满足 ,即模式串有一定长度但远短于文本。

对该过程进行随机性建模,假定可用字符集的规模为 。在文本串与模式串均为随机独立生成的条件下,任意两个单一字符比对成功的概率为 。当要求模式串整体对齐时,连续 个字符均能顺利匹配的先验概率 会呈现指数级衰减,近似等价于 。随着字符集 的增大或模式长度 的增加,该概率值 迅速趋近于 。
这一概率特性说明:大多数对齐位置会在较早字符处失配。因此,优化重点不是加速少数成功匹配,而是低成本处理大量失配,并尽量增加下一次尝试的位移。
蛮力匹配
最直接的方案是蛮力匹配:将模式串起点依次与文本串各位置对齐并逐字检查。一旦字符失配,就判定当前对齐无效,将模式串右移一位并重新比较。

蛮力算法的第一种实现形式采用显式的双指针并行递增架构。代码如下:
1 | // 返回模式串 P 在文本串 T 中首次出现的位置;未匹配则返回 -1。 |

变量 和 分别是文本串 与模式串 的游标。若 T[i] == P[j],二者同步前移;若失配,则执行 i = i - j + 1,将文本游标回到本轮起点的下一位,并将模式串游标 归零。该回退机制在大量局部匹配场景中会导致重复读取。
第二种实现形式使用标准的双层嵌套循环,外层负责穷举对齐点,内层负责验证。
1 | // 双层循环形式的蛮力匹配。 |

外层 for 将起始对齐位置 限定在 ,避免模式串越界;若模式串长于文本串,则提前返回 -1,避免无符号整数下溢。内层通过偏移 访问文本串,失配时 break 并进入下一对齐位置。该写法没有显式回退文本指针,但计算量与第一种实现等效。
时空复杂度
蛮力策略只依赖标量游标,空间复杂度为 。时间复杂度取决于字符序列:最优情况下,模式串首字符很少匹配文本,内层比较很快结束,整体接近 ;最坏情况下,文本和模式串存在大量相同前缀,只在末尾失配,每次对齐都会重复扫描较长片段,复杂度退化为 。在大规模文本流中,文本指针回退还会破坏顺序读取。

KMP 算法
状态不变性
要突破二次复杂度,需要利用扫描过程中的状态不变性。假设某一时刻文本指针 与模式指针 正在比较,且此前 个字符均已匹配,则有区间等式:。
该不变量表示:算法已经知道文本片段 与模式串前缀 完全相同。蛮力算法的问题在于,失配后丢弃了这段已知信息,并从下一个对齐位置重新试错。

一旦 与 失配,理想策略是保持文本指针 不回退,只在模式串内部寻找新的位置 。该位置需要满足更短前缀 可以对接当前已知文本后缀。由于已知文本后缀等价于模式串后缀,问题转化为模式串内部前后缀性质的预处理。
为避免 回退,匹配成功时双指针同步右移;在 处失配时,不复位 ,而是将模式串指针 更新为更小的 ,然后比较 与新的 。

状态变量 可以只根据模式串前缀性质离线预处理得到。无论失配发生在位置 ,都可查询一维状态转移表 next[0, m) 获取新位置。next[j] 表示第 位失配时模式串应跳转到的有效状态。该表用额外空间减少重复扫描,使模式串在失配时可一次右移多个位置。
next[j] 不是“下一位要比较谁”,而是“当模式串第 j 位失配时,已经匹配的前 j 个字符最多还能保留多少个”。文本指针 不回退,模式串指针 回退到 next[j],等价于把模式串整体右移,但不重新读取文本中已经确认过的字符。
状态表实例
以模式串 CHINCHILLA 为例说明 next 数组构造。
针对长度为 10 的字符集,各前缀的转移状态记录如下:
| 索引 j | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 |
|---|---|---|---|---|---|---|---|---|---|---|
| 字符 | C | H | I | N | C | H | I | L | L | A |
跳转状态 next[j] |
-1 | 0 | 0 | 0 | 0 | 1 | 2 | 3 | 0 | 0 |

在该实例中,当 (字符 L)失配时,其前置子串 CHINCHI 包含长度为 3 的相同前后缀 CHI,因此 next = 3。该表用于确定回退位置,并尽量增大模式串位移。
KMP 算法依赖预先构建的 next 数组。主干如下:
1 | // KMP 主匹配过程:文本指针 i 不回退,模式串指针 j 按 next 表回退。 |
程序先调用 buildNext 建立转移表。主 while 循环中,0 > j 用于处理边界状态。当模式串指针退回 -1 时,表示模式串首字符已失配,且没有可用前后缀可复用;此时进入 if 分支,使 回到 ,同时 前进一位。循环结束后释放 next 数组;只有 时返回起点 ,否则返回 -1。
由于文本指针不回退,KMP 适合顺序数据流或不便回放的文本缓冲区。
主循环的进度可以看成由 驱动。成功匹配时 前进;第一个字符都不匹配时,j == -1 也会推动 前进;中间失配时只有 后退, 留在原地等待新的模式串状态来比较同一个文本字符。于是文本串从左到右只被扫描一遍。
模式串与 DFA
KMP 可将每个模式串等效编译为一个确定性有限自动机(DFA)。模式串决定状态机结构,状态数为模式串长度加一,转移由输入字符控制。
若不采用查表方式,也可以将特定模式串(如 chinchilla)转译为分支代码,用 goto 模拟状态转移。逻辑结构如下:

在这种特化的指令集流中,文本指针 在每一步只执行单调递增。若当前字符匹配,指令顺序进入下一状态代码块;若失配,则依据预先计算好的最长公共前后缀边界,通过硬编码的 goto 指令将控制流发配至上游的特定代码块。这证实了模式串预处理实际上是一种针对有限状态转换矩阵的内联优化过程。

失效函数
KMP 的关键在于准确构造状态数组。定义位置 的边界集合 :它包含所有使模式串前缀 与以 结尾的后缀 相等的非负长度 :。

该集合表示所有合法重合长度。为避免漏过潜在匹配,需要选择其中最大元素作为回退位置:。边界条件设为 。
若直接用朴素方法计算所有前缀的边界集合,复杂度较高。由于 依赖字符串线性延续性,可以利用递归传递关系优化。
假定已知当前序列在第 位的最长边界为 。这意味着 。随后考察下一位字符。如果 ,那么原有的前后缀相等关系可以向右无缝延伸一个单位。此时必然得出递推等式:。

若 ,需要寻找更短的候选边界。由于已有 ,寻找当前序列更短边界等价于在前缀串 内寻找最大边界。因此将候选长度缩短为 ,并继续比较新的 与 。该过程称为模式串的自匹配。
构造 next 时,变量 j 相当于“正在扩展的后缀末端”,变量 t 相当于“当前候选前缀长度”。如果 P[j] 和 P[t] 能接上,边界长度就增加;如果接不上,就把候选长度缩短为 next[t]。这和匹配阶段失配时缩短 j 是同一套思想,只不过此时文本串换成了模式串自己。
依据自匹配实效函数构造算法如下:
1 | // 构造 KMP 失效函数 next;N[j] 表示 P[j] 失配时应回退到的状态。 |
迭代中, 表示当前处理位置, 表示当前候选前缀边界长度。由于 可能为 ,0 > t 用于处理重新起步的边界情况。
以模式串 MAMAMMIA 为例,追踪此迭代过程,未经过优化的原始失效边界计算将生成以下序列特征:
| j | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 |
|---|---|---|---|---|---|---|---|---|
P[j] |
M | A | M | A | M | M | I | A |
| 未优化边界 | -1 | 0 | 0 | 1 | 2 | 3 | 1 | 0 |
上述计算得出的实质是原始的失效函数 。为了进一步消除多余的回退,若查表发现当前字符与回退后的字符相同(即 ),意味着采用该回退必然导致再次失配。于是代码中可以加入判定:若相等,则继承更深层的跳转地址,即将 直接赋值为新位置 所存储的回退值,从而形成一步到位的最终 next 数组。
复杂度分析
KMP 主循环中可能出现多次 next 回退,因此需要证明整体不会退化到 。

定义势函数 ,分析循环分支对 的影响:
- 成功匹配分支: 当 或 成立时,执行语句
i++; j++;。在此状态转移下,新势能等于 。较原势能 ,增量严格为 。 - 失配回退分支: 当触发
j = next[j]时,文本指针 保持不变,且根据底层结构定义,下一跳状态必定小于当前状态()。因此减数项 缩小,导致总体势能 必然增大。由于 为整数,其增量至少为 。
因此,无论执行哪个分支,势函数 都单调递增。初始时 。当算法终止时,文本游标最大为 ,模式串回退最小为 ,所以全局势能上界为:。
由于 每轮至少增加 ,且上界为 ,主 while 循环总轮数为 。同理,构造 next 数组需要 。因此,KMP 总时间复杂度为 。
