【力扣】28.实现strStr()--Python实现_python 实现一个strstr的增强函数,可以使用带可选

【题目描述】
实现 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() 定义相符。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/implement-strstr

【解题思路】
KMP实现字符串匹配，参见我的另一篇博客KMP算法解决字符串匹配
用Python实现的代码如下：

class Solution(object):
    def strStr(self, haystack, needle):
        """
        :type haystack: str
        :type needle: str
        :rtype: int
        """
        if needle=="":
            return 0

        # next表构建
        next=[]
        j=0
        next.append(0)
        for i in range(1, len(needle)):
            while j>0 and needle[i]!=needle[j]:
                j = next[j-1]
            if needle[i] == needle[j]:
                j += 1
            next.append(j)
        
        j=0
        for i in range(len(haystack)):
            while j>0 and haystack[i]!=needle[j]:
                j = next[j-1]
            if haystack[i]==needle[j]:
                j += 1
            if j==len(needle):
                return i-len(needle)+1
        return -1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30