LeetCode 28 实现strStr() python

题目 ：实现 strStr() 函数。

给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回 -1。

示例 1:

输入: haystack = "hello", needle = "ll"
输出: 2

示例 2:

输入: haystack = "aaaaa", needle = "bba"
输出: -1

说明:

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

思路 1：调用库函数

思路 2：暴力法，时间复杂度：O((m-n)n)O((m−n)n)，mm 是主字符串，nn 是模式字符串。

思路 3：KMP 算法

        
        
class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        # lenth=len(needle)
        # if lenth==0:
        #     return 0
        # for i in range(len(haystack)-lenth+1):
        #     if haystack[i:i+lenth]==needle:
        #         return i
        # return -1
        
        
        
        
        if not needle : return 0
        n1 = len(haystack)
        n2 = len(needle)
        if n1 < n2:return -1
        def helper(i):
            haystack_p = i
            needle_q = 0
            while needle_q < n2:
                if haystack[haystack_p] !=  needle[needle_q]:
                    return False
                else:
                    haystack_p += 1
                    needle_q += 1
            return True
        for i in range(n1 - n2 + 1):
            if helper(i):
                return i
        return -1
    
    

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        if not needle : 
            return 0
        _next = [0] * len(needle)

        def getNext(needle, _next):
            _next[0] = -1
            i = 0
            j = -1
            while i < len(needle) - 1:
                if j == -1 or needle[i] == needle[j]:
                    i += 1
                    j += 1
                    _next[i] = j
                else:
                    j = _next[j]
        getNext(needle, _next)
        i = 0
        j = 0
        while i < len(haystack) and j < len(needle):
            if j == -1 or haystack[i] == needle[j]:
                i += 1
                j += 1
            else:
                j = _next[j]
        if j == len(needle):
            return i - j
        return -1