python算法-字符串查找KMP算法

最新推荐文章于 2024-05-29 21:12:37 发布

吕海洋

最新推荐文章于 2024-05-29 21:12:37 发布

阅读量524

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/wuchenlhy/article/details/79218205

python 专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种字符串匹配算法，该算法能够高效地在一个较长的文本字符串中查找指定的模式字符串首次出现的位置。通过使用前缀函数预先计算模式串的部分信息，可以避免在匹配过程中进行不必要的回溯，从而提高搜索效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题：对于一个给定非空的 source 字符串和一个 target 字符串，你应该在 source 字符串中找出 target 字符串出现的第一个位置(从0开始)。如果不存在，则返回 -1。

def patten_str(t, p):
    n = len(t)
    m = len(p)
    pi = compute_prefix_function(p)
    q = 0
    if t == p:
        return 0

    for i in range(n):
        print(p[q], t[i])
        if p[q] == t[i]:
            q = q + 1
        else:
            if pi[q] == 0:
                q = 0
            else:
                q = pi[q] - 1

            if p[q] == t[i]:
                q = q + 1

        if q == m:
            return i - m + 1

    return -1


def compute_prefix_function(p):
    m = len(p)
    pi = [0] * m
    k = 0
    for q in range(1, m):
        if p[k] == p[q]:
            k = k + 1
        else:
            k = 0
            if p[k] == p[q]:
                k = k + 1
        pi[q] = k

    return pi


if __name__ == '__main__':
    t = 'ABCDABDCABDEGF'
    p = 'ABD'

    print(patten_str(t, p))

结果