KMP 手写算法

这是我第三次看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

猜你喜欢

转载自blog.csdn.net/abc1234564546/article/details/129194249