通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

jopen 10年前

一、本文简介

本文的目的是简单明了的讲解KMP算法的思想及实现过程。

网上的文章的确有些杂乱,有的过浅,有的太深,希望本文对初学者是非常友好的。

其实KMP算法有一些改良版,这些是在理解KMP核心思想后的优化。

所以本文重点是讲解KMP算法的核心,文章最后会有涉及一些改良过程。

二、KMP算法简介

KMP算法是字符串匹配算法的一种。它以三个发明者命名,Knuth-Morris-Pratt,起头的那个K就是著名科学家Donald Knuth。

三、KMP算法行走过程

首先我们先定义两个字符串作为示例,被匹配串 S = "abcabdabcabeab",匹配串 T = "abcabeab"。

我们的目标就是确定S中是否包含T。

KMP算法的核心是分析匹配串 T 的特征,看看匹配串 T 能告诉我们什么信息。我把 T 着色,如下

T = "abcabeab"

现在看起来似乎比较明显了,三个着色点都是重复的 "ab",似乎这个 T 能告诉我们它有重复的子串"ab"可以利用。那么它们到底怎么用?先不讲具体怎么用,先走一遍KMP这个过程,但是大家需要留意这个"ab"。

1.

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

我们发现T在匹配成功 "abcab" 后到"e"时和S子串的"d"匹配不成功了。

这个时候我们可以得到的先验就是目前T匹配过的S子串,就是"abcab"。而T本身就是"ab"重复的。所以"abcab"可以直接跳到第二个重复"ab"的位置,因为"abcab"中其它字符串开头不可能产生和T对应的匹配,这是很直观的。

因此T应该直接后移三个位置,并且用第三位的"c"和S刚才不匹配的"d"进行比较。

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

可以发现,S匹配的过程是不会回退的。因此匹配过程是S从头到尾的一遍扫描(中间可能因为匹配成功退出),所以这个查找过程是O(N)的复杂度。

2.

此时的匹配串是 "ab", 它告诉我们目前匹配的是"ab"。这个不像"abcab"出现了重复"ab",所以我们知道包括S[5]的"d"之前的子串是垃圾串。因此跳过S[5]重新开始匹配T。

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

最后完成匹配,可见,这样一次对S的扫描完成了对T的查找。

那么对于机器如何实现?第四节会分析。

四、KMP算法核心解析

对于上面的过程,我们抽离出来的话,问题的根本就是对T串重复情况的一个判定。不管S是什么串,只要对T的构造模式分析清楚就可以完成上述跳转过程。

因此需要一个数组记录这个T的 模式函数。

这里先给出这个T的模式函数

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

1.每个字母对应模式函数的值就是匹配到当前位置 i 后,下一次T开始进行比较的下标。

2.而S的移动长度为 i - F(i-1)。

对应上面两个问题

1.比如上面的匹配到的 "abcab" ,是T匹配到 T[5] 即"e"的时候出错的。那么我们需要查看上一个字符的模式函数值,因为上一个函数值才代表了已经匹配的串。

发现F(4)=2,说明下次比较从T[2]即"c"开始。因为"abcab"有重复"ab",第一个"ab"不需要比较。

2.而S下标移动多少呢?S下标的移动即找到T的初试位置对应的S的下标。对应上面第二张图,S[3]和T的移动对应起来了。3的获得就是通过上面公式 5-F(4)得到的。其实这个结果和1得到匹配位置的思路是一样的,不过又向前移动对齐了开头。

为什么如此构造这个模式函数?

就是因为F值表示了对于位置 i, T[i]有无重复,并且重复下标的位置在哪(F[i])。既然我们获取了重复下标的位置,那么其它的相关值可以推出来了。

基于这个思路,再给出另外两个T串的模式函数值,帮助大家思考。

1.

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

2.

通俗易懂的讲解KMP算法(字符串匹配算法)及代码实现

如何快速构造这个模式函数?

这个留给大家思考一下了,应该也比较直接了,注意是查找重复位置。如果不大明白,可以参考下面的代码。

五、KMP算法实现

 再添加个例子帮助大家思考:

S1 = "aaaaaaaaaaaaaaaab"  S2 = "aaaaafaaaaaaaaab"   T = "aaaab"

对于S1和S2两种情况,应如何匹配。

代码如下:

/*      return val means the begin pos of haystack      -1 means no matching substring  */  int KMP(char *haystack, char *needle) {      // pre-process      if(haystack[0] == 0 && needle[0] == 0)          return 0;        int i, j, k, min, cur;        //construct F(t) in vector len      vector<int > len;      len.push_back(0);      for(i=1; needle[i] != 0; i++){          if(len[i-1] == 0){              if(needle[i] == needle[0])                  len.push_back(1);              else                  len.push_back(0);          }else{              if(needle[i] == needle[len[i-1]])                  len.push_back(len[i-1]+1);              else                  len.push_back(0);          }      }      // KMP finder      j = 0;      for(i=0; haystack[i] != 0; ) {          // matching          for(; needle[j] != 0; j++) {              if(haystack[i+j] != needle[j])                  break;          }          //finded          if(needle[j] == 0)              return i;          else{ // jump              if(j){                  cur = j - len[j-1];                  i += cur;                  j = len[j-1];              }else{                  j = 0;                  i++;              }          }      }      //match failed      return -1;  }

六、KMP算法改进

KMP算法有一些改进版本加速查找,一般可以通过S串中的一些信息加速匹配过程。

比如若  S = "aaaaaaaafaaaaaaaaaaaab", T = "aaaaaaab"。

在查找过程中,S中间的 "f" 起到了阻挡作用。但是由于我们只是考虑T的先验信息,遇到"f" 不匹配会导致T每次后移一步进行新的匹配,直到T的开头碰到了"f"。

但是如果我们加入"f"这个原串S的信息,由于"f" != "b" && "f" != "a" && i-1 = F(i-1) ,所以直接跳到"f"后进行新的匹配会更快速的查找。

但是这些改进都是基于KMP基础算法之上的,因此把握核心要点不仅省时省力,更能有效扩展。

七、参考

[1] 《字符串匹配的KMP算法》 http://www.ruanyifeng.com/blog/2013/05/Knuth%E2%80%93Morris%E2%80%93Pratt_algorithm.html

[2] 算法导论

来自:http://www.cnblogs.com/xiaoboCSer/p/4237941.html