先上代码:

class Solution( ):
    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)

操作很骚,我很佩服……

 

收藏 打印