[数据结构与算法]python 实现KMP算法

class Solution:
    def getNext(self, needle: str):
        # 后缀匹配指向
        i = 0
        # 前缀匹配指向
        j = -1
        # 初始化 next 数组
        next = [-1] * len(needle)

        # 此处 next[0] = -1，所以只需要求剩下的 len(T)-1 个即可
        while i < len(needle) - 1:
            # j == -1 就是找无可找 or 匹配成功，相同前缀长度增加1
            if j == -1 or needle[i] == needle[j]:
                i += 1
                j += 1
                next[i] = j
            # 匹配不成功则在前面的子串中继续搜索，直至找不到（即 j== -1 的情况）
            else:
                j = next[j]

        return next

    def strStr(self, haystack: str, needle: str) -> int:
        i = 0
        j = 0
        next = self.getNext(needle)
        while i < len(haystack) and j < len(needle):
            # j == -1 找无可找，从 S[i+1] 开始和 T[0] 匹配 or 当匹配成功时，往下匹配。
            if j == -1 or haystack[i] == needle[j]:
                i += 1
                j += 1
            # 匹配不成功则用 next(j) 找下一次匹配的位置
            else:
                j = next[j]
        # 如果模式串在主串中存在
        if j == len(needle):
            return i - j
        else:
            return -1