这是我第三次看kmp算法。。这次搞懂了就在这里记录一下,下次忘了可以看看
对于一个字符串的next数组,我引用一下这篇文章哈https://www.acwing.com/solution/content/14666/
现在就是放到代码里要这么求这个next数组呢,要用到两个指针,一个指针就是遍历整个字符串,另一个是处理前缀和的长度
以下代码匹配主串中子串第一个出现的起始点和终止点
def find_next(p):
next = [0] * (len(p)+1)
j,k = 0,-1
next[0] = -1 # 防止死循环 k一直等于0 j也不加
while(j < len(p) - 1):
if (k == -1 or p[j] == p[k]):
j += 1
k += 1
next[j] = k
else:
k = next[k]
next[0] = 0
return next
if __name__ == '__main__':
n = int(input())
p = str(input())
m = int(input())
s = str(input())
next = find_next(p)
i = 0 # 主串中的指针
j = 0 # 子串中的指针
while i < len(s):
if s[i] == p[j]:
i += 1
j += 1
elif j > 0: # 匹配失败,根据next跳过前面的字符串
j = next[j]
else : # 字符串第一个就匹配失败
i = i + 1
if j == len(p):
print(i - j, i-1) # 匹配第一个字符串起始和终点
break