题目描述:
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。
示例 1:
输入:haystack = “sadbutsad”, needle = “sad”
输出:0
解释:“sad” 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。
示例 2:
输入:haystack = “leetcode”, needle = “leeto”
输出:-1
解释:“leeto” 没有在 “leetcode” 中出现,所以返回 -1 。
思路一:
(1)首先将两个字符串的长度分为三种情况讨论。
(2)对于haystack长度小于needle的情况,needle 一定不是 haystack 的一部分,直接返回 -1。
(3)对于haystack长度等于needle的情况,再分两种情况,如果haystack不等于needle,直接返回-1,如果相等,直接返回0。
(4)对于haystack长度大于needle的情况,对haystack进行遍历,找到第一个与needle相同的字符段,并返回索引。
代码实现如下:
class Solution(object):
def strStr(self, haystack, needle):
"""
:type haystack: str
:type needle: str
:rtype: int
"""
if len(haystack) < len(needle):
return -1
elif len(haystack) == len(needle):
if haystack == needle:
return 0
else:
return -1
else:
for i in range(len(haystack)):
if needle == haystack[i:i + len(needle)]:
return i
return -1
执行分布用时如下: