目录
image
前言
本文是对下文的小结:https://ethsonliu.com/2018/04/kmp.html。原文基本可以看懂kmp算法,看不懂没关系,看了最后新增的小结部分一定懂.
正文
给定一个主串(以 S 代替)和模式串(以 P 代替),要求找出 P 在 S 中出现的位置,此即串的模式匹配问题。
在继续下面的内容之前,有必要在这里介绍下两个概念:真前缀 和 真后缀。

由上图所得, "真前缀" 指除了自身以外,一个字符串的全部头部组合;"真后缀" 指除了自身以外,一个字符串的全部尾部组合。(网上很多博客,应该说是几乎所有的博客,都是 "前缀"。严格来说,"真前缀" 和 "前缀" 是不同的,既然不同,还是不要混为一谈的好!)
朴素字符串匹配算法
初遇串的模式匹配问题,我们脑海中的第一反应,就是朴素字符串匹配(即所谓的暴力匹配),代码如下:
/* 字符串下标始于 0 */
int NaiveStringSearch(string S, string P)
{
int i = 0; // S 的下标
int j = 0; // P 的下标
int s_len = S.size();
int p_len = P.size();
while (i < s_len && j < p_len)
{
if (S[i] == P[j]) // 若相等,都前进一步
{
i++;
j++;
}
else // 不相等
{
i = i - j + 1;
j = 0;
}
}
if (j == p_len) // 匹配成功
return i - j;
return -1;
}
暴力匹配的时间复杂度为 O(nm),其中 n 为 S 的长度,m 为 P 的长度。很明显,这样的时间复杂度很难满足我们的需求。
接下来进入正题:时间复杂度为 Θ(n+m) 的 KMP 算法。
KMP字符串匹配算法
算法流程
以下摘自阮一峰的字符串匹配的 KMP 算法,并作稍微修改。
(1)

首先,主串 "BBC ABCDAB ABCDABCDABDE" 的第一个字符与模式串 "ABCDABD" 的第一个字符,进行比较。因为 B 与 A 不匹配,所以模式串后移一位。
(2)

因为 B 与 A 又不匹配,模式串再往后移。
(3)

就这样,直到主串有一个字符,与模式串的第一个字符相同为止。
(4)

接着比较主串和模式串的下一个字符,还是相同。
(5)

直到主串有一个字符,与模式串对应的字符不相同为止。
(6)

这时,最自然的反应是,将模式串整个后移一位,再从头逐个比较。这样做虽然可行,但是效率很差,因为你要把"搜索位置"移到已经比较过的位置,重比一遍。
(7)

一个基本事实是,当空格与 D 不匹配时,你其实是已经知道前面六个字符是 "ABCDAB"。KMP 算法的想法是,设法利用这个已知信息,不要把"搜索位置"移回已经比较过的位置,而是继续把它向后移,这样就提高了效率。
(8)
i | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 |
---|---|---|---|---|---|---|---|---|
模式串 | A | B | C | D | A | B | D | '\0' |
next[i] | -1 | 0 | 0 | 0 | 0 | 1 | 2 | 0 |
怎么做到这一点呢?可以针对模式串,设置一个跳转数组 int next[]
,这个数组是怎么计算出来的,后面再介绍,这里只要会用就可以了。
(9)

已知空格与 D 不匹配时,前面六个字符 "ABCDAB" 是匹配的。根据跳转数组可知,不匹配处 D 的 next 值为 2,因此接下来从模式串下标为 2 的位置开始匹配。
(10)

因为空格与 C 不匹配,C 处的 next 值为 0,因此接下来模式串从下标为 0 处开始匹配。
(11)

因为空格与 A 不匹配,此处 next 值为 -1,表示模式串的第一个字符就不匹配,那么直接往后移一位。
(12)

逐位比较,直到发现 C 与 D 不匹配。于是,下一步从下标为 2 的地方开始匹配。
(13)

逐位比较,直到模式串的最后一位,发现完全匹配,于是搜索完成。
next 数组是如何求出的
next 数组的求解基于 "真前缀" 和 "真后缀",即 next[i]
等于 P[0]...P[i - 1]
最长的相同真前后缀的长度(请暂时忽视 i 等于 0 时的情况,下面会有解释)。我们依旧以上述的表格为例,为了方便阅读,我复制在下方了。
i | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 |
---|---|---|---|---|---|---|---|---|
模式串 | A | B | C | D | A | B | D | '\0' |
next[ i ] | -1 | 0 | 0 | 0 | 0 | 1 | 2 | 0 |
- i = 0,对于模式串的首字符,我们统一为
next[0] = -1
; - i = 1,前面的字符串为
A
,其最长相同真前后缀长度为 0,即next[1] = 0
; - i = 2,前面的字符串为
AB
,其最长相同真前后缀长度为 0,即next[2] = 0
; - i = 3,前面的字符串为
ABC
,其最长相同真前后缀长度为 0,即next[3] = 0
; - i = 4,前面的字符串为
ABCD
,其最长相同真前后缀长度为 0,即next[4] = 0
; - i = 5,前面的字符串为
ABCDA
,其最长相同真前后缀为A
,即next[5] = 1
; - i = 6,前面的字符串为
ABCDAB
,其最长相同真前后缀为AB
,即next[6] = 2
; - i = 7,前面的字符串为
ABCDABD
,其最长相同真前后缀长度为 0,即next[7] = 0
。
那么,为什么根据最长相同真前后缀的长度就可以实现在不匹配情况下的跳转呢?举个代表性的例子:假如 i = 6
时不匹配,此时我们是知道其位置前的字符串为 ABCDAB
,仔细观察这个字符串,首尾都有一个 AB
,既然在 i = 6
处的 D 不匹配,我们为何不直接把 i = 2
处的 C 拿过来继续比较呢,因为都有一个 AB
啊,而这个 AB
就是 ABCDAB
的最长相同真前后缀,其长度 2 正好是跳转的下标位置。
有的读者可能存在疑问,若在 i = 5
时匹配失败,按照我讲解的思路,此时应该把 i = 1
处的字符拿过来继续比较,但是这两个位置的字符是一样的啊,都是 B
,既然一样,拿过来比较不就是无用功了么?其实不是我讲解的有问题,也不是这个算法有问题,而是这个算法还未优化,关于这个问题在下面会详细说明,不过建议读者不要在这里纠结,跳过这个,下面你自然会恍然大悟。
思路如此简单,接下来就是代码实现了,如下:
/* P 为模式串,下标从 0 开始 */
void GetNext(string P, int next[])
{
int p_len = P.size();
int i = 0; // P 的下标
int j = -1;
next[0] = -1;
while (i < p_len)
{
if (j == -1 || P[i] == P[j])
{
i++;
j++;
next[i] = j;
}
else
j = next[j];
}
}
一脸懵逼,是不是。。。上述代码就是用来求解模式串中每个位置的 next[]
值。
下面具体分析,我把代码分为两部分来讲:
(1):i 和 j 的作用是什么?
i 和 j 就像是两个”指针“,一前一后,通过移动它们来找到最长的相同真前后缀。
(2):if...else...语句里做了什么?

假设 i 和 j 的位置如上图,由 next[i] = j
得,也就是对于位置 i 来说,区段 [0, i - 1] 的最长相同真前后缀分别是 [0, j - 1] 和 [i - j, i - 1],即这两区段内容相同。
按照算法流程,if (P[i] == P[j])
,则 i++; j++; next[i] = j;
;若不等,则 j = next[j]
,见下图:

next[j]
代表 [0, j - 1] 区段中最长相同真前后缀的长度。如图,用左侧两个椭圆来表示这个最长相同真前后缀,即这两个椭圆代表的区段内容相同;同理,右侧也有相同的两个椭圆。所以 else 语句就是利用第一个椭圆和第四个椭圆内容相同来加快得到 [0, i - 1] 区段的相同真前后缀的长度。
细心的朋友会问 if 语句中 j == -1
存在的意义是何?第一,程序刚运行时,j 是被初始为 -1,直接进行 P[i] == P[j]
判断无疑会边界溢出;第二,else 语句中 j = next[j]
,j 是不断后退的,若 j 在后退中被赋值为 -1(也就是 j = next[0]
),在 P[i] == P[j]
判断也会边界溢出。综上两点,其意义就是为了特殊边界判断。
完整代码
#include <iostream>
#include <string>
using namespace std;
/* P 为模式串,下标从 0 开始 */
void GetNext(string P, int next[])
{
int p_len = P.size();
int i = 0; // P 的下标
int j = -1;
next[0] = -1;
while (i < p_len)
{
if (j == -1 || P[i] == P[j])
{
i++;
j++;
next[i] = j;
}
else
j = next[j];
}
}
/* 在 S 中找到 P 第一次出现的位置 */
int KMP(string S, string P, int next[])
{
GetNext(P, next);
int i = 0; // S 的下标
int j = 0; // P 的下标
int s_len = S.size();
int p_len = P.size();
while (i < s_len && j < p_len) // 因为末尾 '\0' 的存在,所以不会越界
{
if (j == -1 || S[i] == P[j]) // P 的第一个字符不匹配或 S[i] == P[j]
{
i++;
j++;
}
else
j = next[j]; // 当前字符匹配失败,进行跳转
}
if (j == p_len) // 匹配成功
return i - j;
return -1;
}
int main()
{
int next[100] = { 0 };
cout << KMP("bbc abcdab abcdabcdabde", "abcdabd", next) << endl; // 15
return 0;
}
KMP优化
i | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 |
---|---|---|---|---|---|---|---|---|
模式串 | A | B | C | D | A | B | D | '\0' |
next[i] | -1 | 0 | 0 | 0 | 0 | 1 | 2 | 0 |
以 3.2 的表格为例(已复制在上方),若在 i = 5
时匹配失败,按照 3.2 的代码,此时应该把 i = 1
处的字符拿过来继续比较,但是这两个位置的字符是一样的,都是 B
,既然一样,拿过来比较不就是无用功了么?这我在 3.2 已经解释过,之所以会这样是因为 KMP 还未优化。那怎么改写就可以解决这个问题呢?很简单。
/* P 为模式串,下标从 0 开始 */
void GetNextval(string P, int nextval[])
{
int p_len = P.size();
int i = 0; // P 的下标
int j = -1;
nextval[0] = -1;
while (i < p_len)
{
if (j == -1 || P[i] == P[j])
{
i++;
j++;
if (P[i] != P[j])
nextval[i] = j;
else
nextval[i] = nextval[j]; // 既然相同就继续往前找真前缀
}
else
j = nextval[j];
}
}
小结:
1 kmp原理是根据主串滑动一位变成后缀串的特性。子串和主串的比较变成前缀串和后缀串的比较。next数组就是前缀串和后缀串,比较相同字符结果的库.
2 比较过程:把主串想象成xxxxxxxx....的黑盒,比较过程交给计算机,根据返回结果,把主串一个字符一个字符的翻开。
3 最复杂的问题是kmp算法和暴力破解的区别?
暴力破解是怎么略过去的,为什么不继续比较下一个,而是直接到D的位置了?
算法的巧妙就是这点。根据下面的特性和论证的替换就可以理解了。
3.1 特性
特性1: kmp的第一步是,根据子串先做一个前缀和后缀比较的库.
特性2:子串和主串比较实际是,前缀串和主串进行比较。
特性3: 子串ABCDABD和主串比较,到了最后一个D字符不同.说明主串现在是ABCDABX,这时候把主串想象成黑盒, 知道了X之前的字符,是i=6的前缀串,但是不知道X是什么.
特性4: 根据暴力破解向后移动一位,主串变成BCDABX。这个串是以B为后缀的后缀串。即i=6的后缀串.
3.2 论证
(1)子串和主串比较=子串前缀串和主串比较。(特性2)
(2)子串前缀串和主串比较,到了第7个字符D不同,出现了特性3的情况。知道了主串是i=6的前缀串。
(3)根据暴力破解,主串向后移动一位,这时候主串变成i=6的后缀串.
(4) 子串和移动一位的主串比较=子串前缀串和i=6的后缀串比较(这步替换是重点).
(5)子串前缀串和i=6的后缀串已经比较完了,结果放入next(i)中即2。(特性1)
notes:原文写作顺序:简单破解,kmp使用方法,原理,代码. 算法流程的每一步都用图表示。next的计算也是,每一步都写出前缀和后缀,list 所有结果。
网友评论