KMP算法python代码

最新推荐文章于 2025-03-08 18:10:28 发布

原创

最新推荐文章于 2025-03-08 18:10:28 发布 · 451 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#kmp #python #字串匹配

本文介绍如何使用KMP算法解决字符串匹配问题，通过创建next_arr数组优化匹配过程，提高效率。文中还提到了KMP算法的改进，讨论了在字符匹配失败时如何决定继续匹配的起点，并给出了算法的主要逻辑步骤。对于理解有困难的读者，可以参考链接中的详细博客解析。

问题：给定两个字符串a="sdfaabcddsdfssd",b="df"找出字串b在a中的下标位置。

朴素模式匹配算法:

def str_index(a,b,pos=0):
    i = pos
    j = 0
    while i < len(a) and j < len(b):
        if a[i] == b[j]:
            j += 1
            i += 1
        else:
            i = i-j+1
            j = 0
    if j == len(b):
         return i-j
    else:
         return -1

KMP算法:需创建next_arr数组，存放b串中每个下标位置前(不包含当前下标) 字符串的前缀和后缀相同的个数，作用是：在字符匹配失败时，判定应该继续从b串的哪一位字符开始继续匹配


def kmp_next(b):
    next_arr = [-1] * len(b)
    i = 0
    j = -1
    while i < len(b)-1:
        if j == -1 or b[j] == b[i]:
            i += 1
            j += 1
            next_arr[i] = j
        else:
            j = next_arr[j]
    return next_arr

def kmp_index(a,b,pos=0):
    i = pos
    j = -1
    kmp_next_arr = kmp_next(b)
    while i < len(a) and j < len(b)