KMP算法Python

KMP算法Python

题目：在longStr中找到shortStr出现的索引位置。[后面的介绍中longStr用A代替，shortStr用B代替]

一、KMP算法：

对于字符串的匹配直接使用暴力手段进行循环时，A中的索引不断增加[每次失败都会+1]，同样的对于B的索引，如果和A对应位置匹配失败，也会直接回到索引0位置。复杂度O(AB)。

这种简单的做法没有很好的利用每次匹配成功的信息，如果很很好的利用之前的信息，就像DP那样就美滋滋了，那么KMP就出现了。

KMP算法流程：

如果匹配成功那就继续向下匹配，如果匹配失败，从pnext数组中对应的数字代替B中当前的索引，然后接着重复之前的操作

二、pnext数组：

pnext[j] = n 表示j之前的字符串最大相同前缀和后缀的长度，pnext数组中首位为-1。

def KMP(longStr,shortStr):#在longStr字符串中搜索是否存在shortStr字符串
	lengthShort = len(shortStr);lengthLong = len(longStr)
	i, k, m = 0, -1, lengthShort #k就是pnext数组的初值
	pnext = [-1] * m#pnext[j]=n 在j之前的字符串最大公共前后缀长度为n
	while i < m - 1:
		if k == -1 or shortStr[i] == shortStr[k]:
			i, k = i + 1, k + 1
			pnext[i] = k
		else:
			k = pnext[k]#回溯的，如果当前位置不相等，那么就往回看，看看之前的k位置和现在的i位置可以相等吗
	def KMPMain():
		nonlocal pnext,longStr,shortStr
		j, i = 0, 0#对于匹配位置的初值
		n, m = lengthLong, lengthShort
		while j < n and i < m:
			if i == -1 or longStr[j] == shortStr[i]:
				i, j = i + 1, j + 1
			else:
				i = pnext[i]
			if i == m:
				print('成功匹配索引位置:',j - i)
				i = 0
		return -1
	print('短数组:',shortStr)
	print('pnext数组:',pnext)
	return KMPMain()

感谢阅读！！！