KMP算法(28题实现 strStr() 函数)

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。如果不存在，则返回 -1 。

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        next = self.getNext(needle)
        i1 = 0
        i2 = 0
        while i1<len(haystack) and i2<len(needle):
            if haystack[i1] == needle[i2]:
                i1+=1
                i2+=1
            else:
                if next[i2]==-1:
                    i1+=1
                else:
                    i2=next[i2]
        if i2==len(needle):
            return i1-i2
        else:
            return -1
    
    def getNext(self, needle):
        if len(needle)<2:
            return [-1]
        next = [-1] * len(needle)
        next[0] = -1
        next[1] = 0
        // cn是当前位置的最大前缀最后一个字符index+1
        // i是当前位置，i的最大前后缀要比较i-1
        cn = 0
        i = 2
        while i<len(needle):
            if needle[i-1] == needle[cn]:
                next[i] = cn+1
                cn += 1
                i += 1
            else:
                if cn>0:
                    cn = next[cn]
                else:
                    next[i] = 0
                    i+=1
        return next