（原创）详解KMP算法

KMP算法应该是每一本《数据结构》书都会讲的，算是知名度最高的算法之一了，但很惋惜，我大二那年压根就没看懂过~~~

以后也在不少地方也都常常看到讲解KMP算法的文章，看久了好像也知道是怎么一回事，但总感受有些地方本身仍是没有彻底懂明白。这两天花了点时间总结一下，有点小体会，我但愿能够经过我本身的语言来把这个算法的一些细节梳理清楚，也算是考验一下本身有真正理解这个算法。

 

什么是KMP算法：

KMP是三位大牛：D.E.Knuth、J.H.Morris和V.R.Pratt同时发现的。其中第一位就是《计算机程序设计艺术》的做者！！

KMP算法要解决的问题就是在字符串（也叫主串）中的模式（pattern）定位问题。说简单点就是咱们平时常说的关键字搜索。模式串就是关键字（接下来称它为P），若是它在一个主串（接下来称为T）中出现，就返回它的具体位置，不然返回-1（经常使用手段）。

 

首先，对于这个问题有一个很单纯的想法：从左到右一个个匹配，若是这个过程当中有某个字符不匹配，就跳回去，将模式串向右移动一位。这有什么难的？

咱们能够这样初始化：

 

以后咱们只须要比较i指针指向的字符和j指针指向的字符是否一致。若是一致就都向后移动，若是不一致，以下图：

 

 

A和E不相等，那就把i指针移回第1位（假设下标从0开始），j移动到模式串的第0位，而后又从新开始这个步骤：

 

基于这个想法咱们能够获得如下的程序：

1 /**
 2 
 3  * 暴力破大  解法
 4 
 5  * @param ts 主串
 6 
 7  * @param ps 模式串
 8 
 9  * @return 若是找到，返回在主串中第一个字符出现的下标，不然为-1
10 
11  */
12 
13 public static int bf(String ts, String ps) {
14 
15     char[] t = ts.toCharArray();
16 
17     char[] p = ps.toCharArray();
18 
19     int i = 0; // 主串的位置
20 
21     int j = 0; // 模式串的位置
22 
23     while (i < t.length && j < p.length) {
24 
25        if (t[i] == p[j]) { // 当两个字符相同，就比较下一个
26 
27            i++;
28 
29            j++;
30 
31        } else {
32 
33            i = i - j + 1; // 一旦不匹配，i后退
34 
35            j = 0; // j归0
36 
37        }
38 
39     }
40 
41     if (j == p.length) {
42 
43        return i - j;
44 
45     } else {
46 
47        return -1;
48 
49     }
50 
51 }

上面的程序是没有问题的，但不够好！（想起我高中时候数字老师的一句话：我不能说你错，只能说你不对~~~）

若是是人为来寻找的话，确定不会再把i移动回第1位，由于主串匹配失败的位置前面除了第一个A以外再也没有A了，咱们为何能知道主串前面只有一个A？由于咱们已经知道前面三个字符都是匹配的！（这很重要）。移动过去确定也是不匹配的！有一个想法，i能够不动，咱们只须要移动j便可，以下图：

 

上面的这种状况仍是比较理想的状况，咱们最多也就多比较了再次。但假如是在主串“SSSSSSSSSSSSSA”中查找“SSSSB”，比较到最后一个才知道不匹配，而后i回溯，这个的效率是显然是最低的。

 

大牛们是没法忍受“暴力破大   解”这种低效的手段的，因而他们三个研究出了KMP算法。其思想就如同咱们上边所看到的同样：“利用已经部分匹配这个有效信息，保持i指针不回溯，经过修改j指针，让模式串尽可能地移动到有效的位置。”

 

因此，整个KMP的重点就在于当某一个字符与主串不匹配时，咱们应该知道j指针要移动到哪？

 

接下来咱们本身来发现j的移动规律：

 

如图：C和D不匹配了，咱们要把j移动到哪？显然是第1位。为何？由于前面有一个A相同啊：

 

以下图也是同样的状况：

 

能够把j指针移动到第2位，由于前面有两个字母是同样的：

 

至此咱们能够大概看出一点端倪，当匹配失败时，j要移动的下一个位置k。存在着这样的性质：最前面的k个字符和j以前的最后k个字符是同样的。

若是用数学公式来表示是这样的

P[0 ~ k-1] == P[j-k ~ j-1]

这个至关重要，若是以为很差记的话，能够经过下图来理解：

 

弄明白了这个就应该可能明白为何能够直接将j移动到k位置了。

由于:

当T[i] != P[j]时

有T[i-j ~ i-1] == P[0 ~ j-1]

由P[0 ~ k-1] == P[j-k ~ j-1]

必然：T[i-k ~ i-1] == P[0 ~ k-1]

公式很无聊，能看明白就好了，不须要记住。

这一段只是为了证实咱们为何能够直接将j移动到k而无须再比较前面的k个字符。

 

好，接下来就是重点了，怎么求这个（这些）k呢？由于在P的每个位置均可能发生不匹配，也就是说咱们要计算每个位置j对应的k，因此用一个数组next来保存，next[j] = k，表示当T[i] != P[j]时，j指针的下一个位置。

 

不少教材或博文在这个地方都是讲得比较含糊或是根本就一笔带过，甚至就是贴一段代码上来，为何是这样求？怎么能够这样求？根本就没有说清楚。而这里偏偏是整个算法最关键的地方。

1 public static int[] getNext(String ps) {
 2 
 3     char[] p = ps.toCharArray();
 4 
 5     int[] next = new int[p.length];
 6 
 7     next[0] = -1;
 8 
 9     int j = 0;
10 
11     int k = -1;
12 
13     while (j < p.length - 1) {
14 
15        if (k == -1 || p[j] == p[k]) {
16 
17            next[++j] = ++k;
18 
19        } else {
20 
21            k = next[k];
22 
23        }
24 
25     }
26 
27     return next;
28 
29 }

这个版本的求next数组的算法应该是流传最普遍的，代码是很简洁。但是真的很让人摸不到头脑，它这样计算的依据究竟是什么？

好，先把这个放一边，咱们本身来推导思路，如今要始终记住一点，next[j]的值（也就是k）表示，当P[j] != T[i]时，j指针的下一步移动位置。

先来看第一个：当j为0时，若是这时候不匹配，怎么办？

 

像上图这种状况，j已经在最左边了，不可能再移动了，这时候要应该是i指针后移。因此在代码中才会有next[0] = -1;这个初始化。

若是是当j为1的时候呢？

 

显然，j指针必定是后移到0位置的。由于它前面也就只有这一个位置了~~~

 

下面这个是最重要的，请看以下图：

  

 

请仔细对比这两个图。

咱们发现一个规律：

当P[k] == P[j]时，

有next[j+1] == next[j] + 1

其实这个是能够证实的：

由于在P[j]以前已经有P[0 ~ k-1] == p[j-k ~ j-1]。（next[j] == k）

这时候现有P[k] == P[j]，咱们是否是能够获得P[0 ~ k-1] + P[k] == p[j-k ~ j-1] + P[j]。

即：P[0 ~ k] == P[j-k ~ j]，即next[j+1] == k + 1 == next[j] + 1。

这里的公式不是很好懂，仍是看图会容易理解些。

 

那若是P[k] != P[j]呢？好比下图所示：

 

像这种状况，若是你从代码上看应该是这一句：k = next[k];为何是这样子？你看下面应该就明白了。

 

如今你应该知道为何要k = next[k]了吧！像上边的例子，咱们已经不可能找到[ A，B，A，B ]这个最长的后缀串了，但咱们仍是可能找到[ A，B ]、[ B ]这样的前缀串的。因此这个过程像不像在定位[ A，B，A，C ]这个串，当C和主串不同了（也就是k位置不同了），那固然是把指针移动到next[k]啦。

 

有了next数组以后就一切好办了，咱们能够动手写KMP算法了：

1 public static int KMP(String ts, String ps) {
 2 
 3     char[] t = ts.toCharArray();
 4 
 5     char[] p = ps.toCharArray();
 6 
 7     int i = 0; // 主串的位置
 8 
 9     int j = 0; // 模式串的位置
10 
11     int[] next = getNext(ps);
12 
13     while (i < t.length && j < p.length) {
14 
15        if (j == -1 || t[i] == p[j]) { // 当j为-1时，要移动的是i，固然j也要归0
16 
17            i++;
18 
19            j++;
20 
21        } else {
22 
23            // i不须要回溯了
24 
25            // i = i - j + 1;
26 
27            j = next[j]; // j回到指定位置
28 
29        }
30 
31     }
32 
33     if (j == p.length) {
34 
35        return i - j;
36 
37     } else {
38 
39        return -1;
40 
41     }
42 
43 }

和暴力破大  解相比，就改动了4个地方。其中最主要的一点就是，i不须要回溯了。

 

最后，来看一下上边的算法存在的缺陷。来看第一个例子：

 

显然，当咱们上边的算法获得的next数组应该是[ -1，0，0，1 ]

因此下一步咱们应该是把j移动到第1个元素咯：

 

不难发现，这一步是彻底没有意义的。由于后面的B已经不匹配了，那前面的B也必定是不匹配的，一样的状况其实还发生在第2个元素A上。

显然，发生问题的缘由在于P[j] == P[next[j]]。

因此咱们也只须要添加一个判断条件便可：

public static int[] getNext(String ps) {

    char[] p = ps.toCharArray();

    int[] next = new int[p.length];

    next[0] = -1;

    int j = 0;

    int k = -1;

    while (j < p.length - 1) {

       if (k == -1 || p[j] == p[k]) {

           if (p[++j] == p[++k]) { // 当两个字符相等时要跳过

              next[j] = next[k];

           } else {

              next[j] = k;

           }

       } else {

           k = next[k];

       }

    }

    return next;

} 

好了，至此。KMP算法也结束了。

很奇怪，好像不是很难的东西怎么就把我困住这么久呢？

仔细想一想仍是由于本身太浮躁了，之前老是草草应付，不少细节都没弄清楚，就觉得本身懂了。结果就只能是似懂非懂的。要学东西真的须要静下心来。