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()
感谢阅读!!!