先上代码:
class Solution(object):
def strStr(self, haystack, needle):
if needle not in haystack:
return -1
if needle == '':
return 0
m,n = len(haystack),len(needle)
for i in range(m):
if haystack[i] == needle[0]:
piece = haystack[i:i+n]
if piece == needle:
return i
return -1
思路:
先处理一些特殊情况,减少处理较长输入的处理时间,可以提高平均速度。如果 needle 根本不在haystack里,可以直接返回-1了;如果 needle 是个空串,按要求返回0。
遍历haystack字符串,直到当前字符为needle的第一个字符,此时出现了第一个字符的匹配,我们把从该字符开始,长度与needle相同的子串截取出来,判断这个子串是否就是needle。这里用的是切片的方法,piece = haystack[i:i+n]。
更简单的方法,只有一行:
return haystack.find(needle)
操作很骚,我很佩服……