python动画演示算法_利用动画演示搞懂KMP字符串匹配算法

近日被朋友问到了字符串匹配算法，让我想起了大二上学期在一次校级编程竞赛中我碰到同样的问题时，为自己写出了暴力匹配算法而沾沾自喜的经历。

现在想来，着实有点羞愧，于是埋头去学习了一下KMP算法，为了让自己不至于那么快忘记，也希望小伙伴们能从我的理解中收获一点自己的感悟！

文章伴有精心雕琢的动画以便理解。

我们首先来分析一下暴力算法，为鲜花的诞生献上绿叶！

以下文中统一将需要被匹配的字符串(长的那段)称为 待匹配串 ，把用来匹配的字符串(短的那段)称为 模式串 。

暴力匹配算法的思路很简单，就是每一次都首先将待匹配串和模式串的首字母对齐，然后比对是否相同，若相同则继续比对两个串的下一个位置，如果不相同的话就将模式串向右移动一位，然后再重新开始从头匹配，就像下面这样:arrow_down::arrow_down:

从上面的动画我们可以直观的看出来，下面的模式串在匹配失败之后都只会移动一格，傻里傻气的，这就导致它的时间复杂度是$M*N$，其中M是模式串的长度，N是待匹配串的长度。

对于这个时间复杂度，我不满意！它太傻了，不符合我聪明睿智的气质！

那就来分析一下为何它这么傻。我们可以看到，在第一次匹配失败的时候，我们肯定希望它向右移动至少两格，因为 模式串的第一格和第三格都为a，既然第三格已经匹配成功了，那么把第一格对上第三格匹配的位置，那么无疑肯定也是可以成功的 ，我们的算法本该知道并且利用这一点的！但是它没有，它太傻了。

嗯，这么一说，好像是感觉应该是要把它向着动态规划的方向改(即利用已有信息为下一步提供便利)。

PS：字符串问题百分之八十以上都可以使用动态规划思想达到较低的时间复杂度。

我们大都听过一句老话：人啊，贵在有自知之明。

同时我们肯定也听别人说过：人只有深刻的认识了自己，才能找对位置，迅速地向目标前进！

这两句话用在KMP算法中再合适不过了！

KMP算法的核心便在于，模式串对自己的自我认知！

想一想，我们人对自己的认知是如何的：男，19岁，阳光帅气聪明机智，这些自我认知都存放在我的脑袋里面。

那么，模式串对自己的认知应该存放在哪呢？

对，就是next数组里面！字符串没有大脑，所以它需要额外的空间来存储它对自己的认知并籍此作出高效准确的判断。

那么字符串对自己的认知是怎样的呢？其实很容易理解，就是知道自己身上哪些地方是相同的，这样的话在匹配失败之后就能迅速找准下次开始的点。这里是不是有点模糊了？图来！

以上就是KMP算法的动画，如果觉得动画稍微有点快的话可以多观看几次，在这个动画里我还没有放出next数组的部分，只是用拟人化的手法展现出来。希望大家能够理解，为什么第一次匹配失败可以直接移动两格。

是因为模式串中第三格的a，它知道在第一格有与自己相同的字符，并且把这个信息告诉下一格的字符，让它在匹配失败之后直接把第一格的a移动到它的那个位置上去。

我这里为了大家容易理解，只放出了一个字符相同的情况，大家不妨可以扩展想一下，假如，第一格和第三格的a不是一个字符，而是一个字符串呢？怎么？有点打脑壳？图来！

来看看模式串与其对应的next自我认识数组吧。

i

0

1

2

3

4

5

6

next

-1

0

0

0

1

2

3

string

a

b

c

a

b

c

d

不要去在意next数组的第一个为什么是-1，这是为了代码写的方便，暂且就给它当成0.

在动画中，当一个字符发出“直接移动”的语句的时候，其实是告诉后一个字符，如果你匹配失败了的话，就直接移动，同时后一个字符对应的next数组值为0，当后一个字符匹配失败了，就移动 模式串的长度-这个匹配失败的字符对应的next值 个长度。

从第四个字符(i=3)起，它们都在不断告诉后面一个字符：“将i=0移动到i=3的位置”，这句话对于i=4的字符来说，是移动 4-1 格， 对于i=5的字符来说，是移动 5-2 格，对于i=6的字符来说，是移动 6-3格 ：后面那个减数恰好就是这个字符对应的next数组的值!

因为模式串足够了解自己，所以它能够在匹配失败的时候不用回退，不用每次只移动一格，而是跟随着待匹配串一起移动。待匹配字符串的指针从未回退过，以线性的速度向前一步步越进。

最终：KMP算法的时间复杂度是$M+N$

这里我们不禁发出了感叹！原来认识自己真的这么重要啊！

接下来是求出给定模式串的next数组：

python3代码奉上:arrow_down::arrow_down:

def get_next_lst(ss: str) -> list:

length = len(ss)

next_lst = [0 for _ in range(length)]

next_lst[0] = -1

i = 0

j = -1

while i < length - 1:

if j == -1 or ss[i] == ss[j]:

i += 1

j += 1

next_lst[i] = j

else:

j = next_lst[j]

return next_lst

这段代码最难理解的就是 j=next_lst[j] 这句话，其实这句话也是动态规划的一个思想，看我为你剖析一下。

已知蓝色区域相等且长度都为len，那么很明显， next[i] == len ，若此时模式串 pattern[i] != pattern[j] (两个灰色区域不相等)。那么看下图：

若此时 next[j] == len(粉色部分) 那么 S1==S2 ，又因为 next[i] == next[j] ，所以 S1==S3 且 S3 == S4 ，则可以推出 S1 == S4 ，这样我们就利用前面所获得的信息，推出了 S1 == S4 这个信息，然后将J移动到S1后一格，只要再次比较 patter[i] 与 patter[j] 的相等情况，就可以得出 next[i+1] 的值。这里因为i始终向后移动，所以也是线性时间复杂度的算法。

ohhhhhhhhh～

到这里，大家就明白了为啥KMP算法的时间复杂度是$M+N$了。

KMP匹配字符串的完整代码附上！

class KMP():

def __init__(self, ss: str) -> list:

self.length = len(ss)

self.next_lst = [0 for _ in range(self.length)]

self.next_lst[0] = -1

i = 0

j = -1

while i < self.length - 1:

if j == -1 or ss[i] == ss[j]:

i += 1

j += 1

self.next_lst[i] = j

else:

j = self.next_lst[j]

self.pattern = ss

def match_str(self, ss:str):

ans_lst = []

j = 0

for i in range(len(ss)):

if ss[i] != self.pattern[j]:

j = self.next_lst[j] if self.next_lst[j] != -1 else 0

if ss[i] == self.pattern[j]:

j += 1

if j == self.length:

return i + 1 - self.length

return -1

tmp_kmp = KMP('iabc')

print(tmp_kmp.match_str('adosjfoiajsoifjasiofjoiasdjoiabc'))

看到这里，如果你觉得这篇文章对你理解KMP算法有帮助的话呢，不妨关注我，我会持续更新各种有用的东西。我的个人公众号是【程序小员】，也欢迎你的关注哦！

我是落阳，谢谢你的到访～