## KMP 算法(Knuth-Morris-Pratt) > 如果阅读时,发现错误,或者动画不可以显示的问题可以添加我微信好友 **[tan45du_one](https://raw.githubusercontent.com/tan45du/tan45du.github.io/master/个人微信.15egrcgqd94w.jpg)** ,备注 github + 题目 + 问题 向我反馈 > > 感谢支持,该仓库会一直维护,希望对各位有一丢丢帮助。 > > 另外希望手机阅读的同学可以来我的 [**公众号:程序厨**](https://raw.githubusercontent.com/tan45du/test/master/微信图片_20210320152235.2pthdebvh1c0.png) 两个平台同步,想要和题友一起刷题,互相监督的同学,可以在我的小屋点击[**刷题小队**](https://raw.githubusercontent.com/tan45du/test/master/微信图片_20210320152235.2pthdebvh1c0.png)进入。 我们刚才讲了 BM 算法,虽然不是特别容易理解,但是如果你用心看的话肯定可以看懂的,我们再来看一个新的算法,这个算法是考研时必考的算法。实际上 BM 和 KMP 算法的本质是一样的,你理解了 BM 再来理解 KMP 那就是分分钟的事啦。 我们先来看一个实例 ![](https://img-blog.csdnimg.cn/20210319193924180.gif) 为了让读者更容易理解,我们将指针移动改成了模式串移动,两者相对与主串的移动是一致的,重新比较时都是从指针位置继续比较。 通过上面的实例是不是很快就能理解 KMP 算法的思想了,但是 KMP 的难点不是在这里,不过多思考,认真看理解起来也是很轻松的。 在上面的例子中我们提到了一个名词,**最长公共前后缀**,这个是什么意思呢?下面我们通过一个较简单的例子进行描述。 ![KMP例子](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/KMP例子.1uirbimk5fcw.png) 此时我们在红色阴影处匹配失败,绿色为匹配成功部分,则我们观察匹配成功的部分。 我们来看一下匹配成功部分的所有前缀 ![在这里插入图片描述](https://img-blog.csdnimg.cn/20210401204019428.png) 我们的最长公共前后缀如下图,则我们需要这样移动 ![原理](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/原理.bghc3ecm4z4.png) 好啦,看完上面的图,KMP 的核心原理已经基本搞定了,但是我们现在的问题是,我们应该怎么才能知道他的最长公共前后缀的长度是多少呢?怎么知道移动多少位呢? 刚才我们在 BM 中说到,我们移动位数跟主串无关,只跟模式串有关,跟我们的 bc,suffix,prefix 数组的值有关,我们通过这些数组就可以知道我们每次移动多少位啦,其实 KMP 也有一个数组,这个数组叫做 next 数组,那么这个 next 数组存的是什么呢? next 数组存的咱们最长公共前后缀中,前缀的结尾字符下标。是不是感觉有点别扭,我们通过一个例子进行说明。 ![next数组](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/next数组.3nir7pgcs9c0.png) 我们知道 next 数组之后,我们的 KMP 算法实现起来就很容易啦,另外我们看一下 next 数组到底是干什么用的。 ![KMP1](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/KMP1.j74ujxjuq1c.png) ![kmp2](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/kmp2.6jx846nmyd00.png) 剩下的就不用说啦,完全一致啦,咱们将上面这个例子,翻译成和咱们开头对应的动画大家看一下。 ![请添加图片描述](https://img-blog.csdnimg.cn/20210319193924754.gif) 下面我们看一下代码,标有详细注释,大家认真看呀。 **注:很多教科书的 next 数组表示方式不一致,理解即可** ```java class Solution { public int strStr(String haystack, String needle) { //两种特殊情况 if (needle.length() == 0) { return 0; } if (haystack.length() == 0) { return -1; } // char 数组 char[] hasyarr = haystack.toCharArray(); char[] nearr = needle.toCharArray(); //长度 int halen = hasyarr.length; int nelen = nearr.length; //返回下标 return kmp(hasyarr,halen,nearr,nelen); } public int kmp (char[] hasyarr, int halen, char[] nearr, int nelen) { //获取next 数组 int[] next = next(nearr,nelen); int j = 0; for (int i = 0; i < halen; ++i) { //发现不匹配的字符,然后根据 next 数组移动指针,移动到最大公共前后缀的, //前缀的后一位,和咱们移动模式串的含义相同 while (j > 0 && hasyarr[i] != nearr[j]) { j = next[j - 1] + 1; //超出长度时,可以直接返回不存在 if (nelen - j + i > halen) { return -1; } } //如果相同就将指针同时后移一下,比较下个字符 if (hasyarr[i] == nearr[j]) { ++j; } //遍历完整个模式串,返回模式串的起点下标 if (j == nelen) { return i - nelen + 1; } } return -1; } //这一块比较难懂,不想看的同学可以忽略,了解大致含义即可,或者自己调试一下,看看运行情况 //我会每一步都写上注释 public int[] next (char[] needle,int len) { //定义 next 数组 int[] next = new int[len]; // 初始化 next[0] = -1; int k = -1; for (int i = 1; i < len; ++i) { //我们此时知道了 [0,i-1]的最长前后缀,但是k+1的指向的值和i不相同时,我们则需要回溯 //因为 next[k]就时用来记录子串的最长公共前后缀的尾坐标(即长度) //就要找 k+1前一个元素在next数组里的值,即next[k+1] while (k != -1 && needle[k + 1] != needle[i]) { k = next[k]; } // 相同情况,就是 k的下一位,和 i 相同时,此时我们已经知道 [0,i-1]的最长前后缀 //然后 k + 1 又和 i 相同,最长前后缀加1,即可 if (needle[k+1] == needle[i]) { ++k; } next[i] = k; } return next; } } ``` Python Code: ```python from typing import List class Solution: def strStr(self, haystack: str, needle: str)->int: # 两种特殊情况 if len(needle) == 0: return 0 if len(haystack) == 0: return -1 # 长度 halen = len(haystack) nelen = len(needle) # 返回下标 return self.kmp(haystack, halen, needle, nelen) def kmp(self, hasyarr: str, halen: int, nearr: str, nelen: int)->int: # 获取next 数组 next = self.next(nearr, nelen) j = 0 for i in range(0, halen): # 发现不匹配的字符,然后根据 next 数组移动指针,移动到最大公共前后缀的, # 前缀的后一位,和咱们移动模式串的含义相同 while j > 0 and hasyarr[i] != nearr[j]: j = next[j - 1] + 1 # 超出长度时,可以直接返回不存在 if nelen - j + i > halen: return -1 # 如果相同就将指针同时后移一下,比较下个字符 if hasyarr[i] == nearr[j]: j += 1 # 遍历完整个模式串,返回模式串的起点下标 if j == nelen: return i - nelen + 1 return -1 # 这一块比较难懂,不想看的同学可以忽略,了解大致含义即可,或者自己调试一下,看看运行情况 # 我会每一步都写上注释 def next(self, needle: str, len:int)->List[int]: # 定义 next 数组 next = [0] * len # 初始化 next[0] = -1 k = -1 for i in range(1, len): # 我们此时知道了 [0,i-1]的最长前后缀,但是k+1的指向的值和i不相同时,我们则需要回溯 # 因为 next[k]就时用来记录子串的最长公共前后缀的尾坐标(即长度) # 就要找 k+1前一个元素在next数组里的值,即next[k+1] while k != -1 and needle[k + 1] != needle[i]: k = next[k] # 相同情况,就是 k的下一位,和 i 相同时,此时我们已经知道 [0,i-1]的最长前后缀 # 然后 k + 1 又和 i 相同,最长前后缀加1,即可 if needle[k + 1] == needle[i]: k += 1 next[i] = k return next ```