实现 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() 定义相符。
#最直观的思路
首先判断原str是否为空,如果为空,
则判断我们需要找的结构体是否也为空,若为空则直接返回0,反之则返回-1.(此处先处理一下特殊情况)
随后又比较两个str的长度,如果原str小于needle,则直接返回-1.
最后开始正常的比较:从位置0开始,逐一比较。直接找到相等的str返回0
反之则返回-1
class Solution:
def strStr(self, haystack: str, needle: str) -> int:
if haystack == "":
if needle == "":
return 0
else:
return -1
else:
if len(haystack) - len(needle) < 0:
return -1
for i in range (len(haystack) - len(needle) + 1):
if haystack[i:i+len(needle)] == needle:
return i
break
if i == len(haystack) - len(needle):
return -1
break
#更简单的写法
class Solution:
def strStr(self, haystack: str, needle: str) -> int:
l1 = len(haystack)
l2 = len(needle)
i = 0
while i <= l1 - l2:
if haystack[i:i+l2] == needle:
return i
i += 1
return -1