【Day 83 】2024-06-29 - 28 实现 strStr(

[azl397985856]

28 实现 strStr(

入选理由

暂无

题目地址

[ 之 BF&RK 篇）

https://leetcode-cn.com/problems/implement-strstr/]( 之 BF&RK 篇）

https://leetcode-cn.com/problems/implement-strstr/)

前置知识

• 滑动窗口
• 字符串
• Hash 运算

  题目描述

  
  实现 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() 定义相符。

[lxy1108]

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        # return haystack.find(needle)
        for i in range(len(haystack)):
            flag = True
            for j in range(len(needle)):
                if i+j>=len(haystack) or needle[j]!=haystack[i+j]:
                    flag=False
                    break
            if flag:
                return i
        return -1

[CathyShang]

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        ans = -1
        n,m = len(haystack), len(needle)
        if n<m: return ans
        for i in range(n-m+1):
            if haystack[i:i+m]==needle:
                ans = i
                break
        return ans

[zhiyuanpeng]

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        """
        s[i], s[i+1], ..., s[i+m-1]
        h[i] = s[i]*base^(m-1) + s[i+1]*base^(m-2) + ... + s[i+m-1]
        h[i+1] = s[i+1]*base^(m-1) + ... + s[i+m-1]*base + s[i+m]
               = h[i]*base - s[i]*base^m + s[i+m]
        max_base = base^m (not base^(m-1))
        """
        n = len(haystack)
        m = len(needle)
        b = 26
        mod = 1_000_000_033
        if n < m:
            return -1
        max_base = 1
        for _ in range(m):
            max_base = (max_base*b) % mod

        def hashf(string):
            v = 0
            for i in range(m):
                v = (v*b + (ord(string[i]) - ord("a"))) % mod
            return v % mod

        hash_val = hashf(needle)
        for i in range(n - m + 1):
            if i == 0:
                val = hashf(haystack[i: i+m])
            else:
                val = (val * b - (ord(haystack[i-1]) - ord("a")) * max_base + (ord(haystack[i + m - 1]) - ord("a")) + mod) % mod
            if val == hash_val:
                if all(needle[j] == haystack[i + j] for j in range(m)):
                    return i
        return -1