28. 实现 strStr()

28. 实现 strStr()

1.题目

2.我的解决方案

• 不出我所料，超时了

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        if not needle:
            return 0
        
        i = 0
        j = 0

        while i < len(haystack):
            while i < len(haystack) and haystack[i] != needle[j]:
                i += 1
            
            tmp = i

            while i < len(haystack) and j < len(needle) and haystack[i] == needle[j]:
                i += 1
                j += 1
            
            if j == len(needle):
                return tmp
            else:
                i = tmp + 1
                j = 0
        
        return -1

3.官方的解决方案

• KMP算法
• KMP算法的官方视频：
• 视频一
• 视频二

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        a = len(needle)
        b = len(haystack)
        if a == 0:    # 空串返回0
            return 0
        next = self.getnext(a, needle)   # 创建next数组
        p = -1  # 指向模式串的指针
        for j in range(b):
            while p >= 0 and needle[p+1]!=haystack[j]:  # 和主串不匹配时，连续回退
                p = next[p]
            if needle[p+1] == haystack[j]:  # 和主串匹配时，模式串指针前移
                p += 1
            if p == a-1:    # 匹配成功
                return j-a+1
        return -1   # 主串遍历完但是没有找到

    """
    构造next数组只需要关注模式串
    需要完成3个任务：
    1.next数组初始化
    2.前缀和后缀匹配处理
    3.前缀和后缀不匹配处理
    """
    def getnext(self, a, needle):   
        next=['' for i in range(a)] # 创建next数组空间
        k = -1  # next初始化，0号用-1进行初始化
        next[0] = k # k直线最长公共前后缀中的前缀末尾，同时表示公共字符串的长度
        for i in range(1, len(needle)): # i指向的是公共最长前后缀中后缀的末尾，因为前后缀要比较，所以i的初始值为1
            while (k>-1 and needle[k+1]!=needle[i]): # 当前缀和后缀不匹配时，前缀末尾k应该回到next[k]，这个是循环的回退
                k = next[k]
            if needle[k+1] == needle[i]:    # 前缀和后缀相等的时候
                k += 1  # 公共前缀的长度+1
            next[i] = k # i的回退最终确定就是k
        return next