题目

实现一个子串匹配算法。

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

示例 1:

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

示例 2:

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

说明:

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

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

解答

def strStr(haystack: str, needle: str) -> int:
    '''字符串匹配算法有多种, 此处实现的是最简单的, 同时也是时间复杂度最大的
    '''
    if not haystack and not needle:
        return 0
    elif not haystack:
        return -1
    elif not needle:
        return 0

    if len(needle) > len(haystack):
        return -1

    for i in range(len(haystack)):
        if needle[0] != haystack[i]:  # 当前位置元素和needle第一个元素不同
            continue
        elif len(needle) == 1:  # needle 长度等于1, 放在elif是为了循环时减少一次判断
            return i

        if len(haystack) - i < len(needle):  # 剩下的元素小于 needle 的长度
            break

        for j in range(1, len(needle)):
            if needle[j] != haystack[i + j]:  # 第j个元素是否相同
                break
            elif j == len(needle) - 1:
                return i

    return -1