题目描述:
给定一个 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:
字符串的逐一滑动比较
代码1:
class Solution:
def strStr(self, haystack, needle):
m, n = len(needle), len(haystack)
for start in range(n - m + 1):
if haystack[start: start + m] == needle:
return start
return -1
解题思路2: 双指针
来源:力扣
上一个方法的缺陷是会将 haystack
所有长度为 L 的子串都与 needle
字符串比较,实际上是不需要这么做的。
首先,只有子串的第一个字符跟 needle
字符串第一个字符相同的时候才需要比较。
其次,可以一个字符一个字符比较,一旦不匹配了就立刻终止。
如下图所示,比较到最后一位时发现不匹配,这时候开始回溯。需要注意的是,pn 指针是移动到 pn = pn - curr_len + 1
的位置,而不是 pn = pn - curr_len
的位置。
这时候再比较一次,就找到了完整匹配的子串,直接返回子串的开始位置 pn - L
代码2:
class Solution:
def strStr(self, haystack, needle):
m, n = len(needle), len(haystack)
if m == 0:
return 0
pn = 0
while pn < n - m + 1:
while pn < n - m + 1 and haystack[pn] != needle[0]:
pn += 1
curr_len = pL = 0
while pL < m and pn < n and haystack[pn] == needle[pL]:
pn += 1
pL += 1
curr_len += 1
if curr_len == m: return pn - m
pn = pn - curr_len + 1
return -1
参考链接:
https://leetcode-cn.com/problems/implement-strstr/solution/shi-xian-strstr-by-leetcode/
题目来源: