KMP算法基础与优化

最新推荐文章于 2024-07-18 17:11:08 发布

阳光素描

最新推荐文章于 2024-07-18 17:11:08 发布

阅读量142

点赞数

文章标签： KMP

原文链接：https://segmentfault.com/a/1190000008575379

版权

KMP字符串匹配算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

一、背景

给定一个主串（以 S 代替）和模式串（以 P 代替），要求找出 P 在 S 中出现的位置，此即串的模式匹配问题。

Knuth-Morris-Pratt 算法（简称 KMP）是解决这一问题的常用算法之一，这个算法是由高德纳（Donald Ervin Knuth）和沃恩·普拉特在1974年构思，同年詹姆斯·H·莫里斯也独立地设计出该算法，最终三人于1977年联合发表。

在继续下面的内容之前，有必要在这里介绍下两个概念：真前缀和真后缀。
在这里插入图片描述
由上图所得， "真前缀"指除了自身以外，一个字符串的全部头部组合；"真后缀"指除了自身以外，一个字符串的全部尾部组合。（网上很多博客，应该说是几乎所有的博客，也包括我以前写的，都是“前缀”。严格来说，“真前缀”和“前缀”是不同的，既然不同，还是不要混为一谈的好！）

二、KMP字符串匹配基本算法

1.算法流程

以下摘自阮一峰的字符串匹配的KMP算法，并作稍微修改。
（1）
在这里插入图片描述
首先，主串"BBC ABCDAB ABCDABCDABDE"的第一个字符与模式串"ABCDABD"的第一个字符，进行比较。因为B与A不匹配，所以模式串后移一位。

（2）
在这里插入图片描述
因为B与A又不匹配，模式串再往后移。

（3）
在这里插入图片描述
就这样，直到主串有一个字符，与模式串的第一个字符相同为止。

（4）在这里插入图片描述
接着比较主串和模式串的下一个字符，还是相同。
（5）

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

（6）
在这里插入图片描述
这时，最自然的反应是，将模式串整个后移一位，再从头逐个比较。这样做虽然可行，但是效率很差，因为你要把"搜索位置"移到已经比较过的位置，重比一遍。

（7）在这里插入图片描述
一个基本事实是，当空格与D不匹配时，你其实是已经知道前面六个字符是"ABCDAB"。KMP算法的想法是，设法利用这个已知信息，不要把"搜索位置"移回已经比较过的位置，而是继续把它向后移，这样就提高了效率。

（8）
在这里插入图片描述
怎么做到这一点呢？可以针对模式串，设置一个跳转数组int next[]，这个数组是怎么计算出来的，后面再介绍，这里只要会用就可以了。

（9）
在这里插入图片描述已知空格与D不匹配时，前面六个字符"ABCDAB"是匹配的。根据跳转数组可知，不匹配处D的next值为2，因此接下来从模式串下标为2的位置开始匹配。

（10）
在这里插入图片描述
因为空格与Ｃ不匹配，C处的next值为0，因此接下来模式串从下标为0处开始匹配。

（11）
在这里插入图片描述
因为空格与A不匹配，此处next值为-1，表示模式串的第一个字符就不匹配，那么直接往后移一位。

（12）
在这里插入图片描述
逐位比较，直到发现C与D不匹配。于是，下一步从下标为2的地方开始匹配。

（13）
在这里插入图片描述
逐位比较，直到模式串的最后一位，发现完全匹配，于是搜索完成。

2.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正好是跳转的下标位置。

三、全部代码实现

#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 - 1)
    {
        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)
    {
        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 = 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 - 1)
    {
        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];
    }
}