LeetCode--28.strStr() （个人解题思路）

题目内容

实现 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() 定义相符。

个人解题思路

1. 首先，判断这道问题的特殊情况：haystack与needle，这两个字符串都为空时，返回值应为 0；haystack字符串不为空但needle字符串为空时，返回值也应该为 0；haystack字符串为空但needle字符串非空时，返回值应为 -1。把握这三种特殊情况，判断特殊情况代码如下：

    if len(haystack)==0 or len(needle) == 0:
        if len(haystack)==len(needle) or len(needle)==0:
            return 0
        else: 
            return -1

2. 排除了特殊情况后，我们需要考虑用什么样的方式通过needle字符串的内容，逐步对haystack字符串进行扫描。在扫描过程中，若我们可以从haystack字符串中完全匹配到needle字符串，返回needle字符串在haystack中首字符的索引；若haystack没有匹配到needle字符串则返回值应为 -1。针对于对字符串进行扫描，我画了如下思路草图。

思路草图

如图所示，这里我采用了题目中示例1作为参考。我解决问题的思路是设置了两个指针，两个指针分别指向了haystack字符串与needle字符串首字符的位置（图中，haystack = ”hello“，needle = ”ll“）。

位置①

这一部分的图像表示我在一开始进行字符串扫描时的相对位置。如图中，haystack[left] = ‘h’ 而 needle = ‘l’，那么此时我们应该保持right指针仍指在needle的首字符位置（right这个指针认死理，只有我首字符匹配到，它才会开闸放水进而向后移动）；再来看left，既然我当下指向的字符不能让right老兄移动，那我就向后移动吧…这一部分有一个简单的逻辑代码：

if haystack[left] == needle[right]:
    #一旦left与right指向的字符相同就达成共识，进而验证后面的字符。
    left += 1
    right += 1
else:
	#如果left指向的字符没有让right老兄满意，就只能独自前行
	#（这里逻辑还不完整，如果left与right中途无法匹配，反水了那我们该如何定义left与right的位置？）
    left += 1

位置②

这一部分的图像表示left老兄经过两次的移动，终于让right老兄满意了，俩指针达成一致都可以快乐地向后移动了，但这哥俩也有会反水的时候，如下图所示：

现在我的needle[1] 变成了 ”a“，right肯定就寻思，left你不讲究，咱俩之前还说好一样的，现在你变了那我还是得回到我最初的起点（right = 0），left也伤心呀，我也想和right继续开心的做好朋友，那我也回到和right开始达成一致的位置吧（left思考了下，得稍微进步一下，上次的错误不能再犯，由此 left = left - right + 1）有了这一部分的考虑，这一部分的代码逻辑如下：

if haystack[left] == needle[right]:
    left += 1
    right += 1
else:
    left = left - right + 1  #left回退到与right达成共识的下一个位置，从新开始与right协商
    right = 0

终止程序的判断条件

我们已经构思好了left和right两兄弟的移动逻辑与方式，接下来haystack和needle这俩字符串到底能不能满足题目的要求，终究还是由left和right说的算的。如果left老兄都走到haystack尽头（这个尽头，为了方便理解这里就先定义为haystack字串的长度吧，完整代码里对尽头设置有讨论）也没找到满足的条件那肯定就（返回 -1）Say拜拜了~，咦？满足条件？什么条件？此时right老兄登场，如果right走到了needle的尽头那肯定就满足了题目呀！（此时要返回哥俩握手时的位置，这个位置？啊！返回left减去needle字串的长度，没错了！）这一部分代码逻辑如下：

    while left < len(haystack):
        if haystack[left] == needle[right]:
            if right+1 == len(needle):
                return left-len(needle)+1
            left += 1
            right += 1
        else:
            left = left - right + 1
            right = 0
    return -1

完整代码

def strStr(haystack,needle):
    if len(haystack)==0 or len(needle) == 0:
        if len(haystack)==len(needle) or len(needle)==0:
            return 0
        else:
            return -1
    left = 0
    right = 0
    while left < len(haystack):
        if haystack[left] == needle[right]:
            if right+1 == len(needle):
                return left-len(needle)+1
            left += 1
            right += 1
        else:
            left = left - right + 1
            right = 0
    return -1

总结

这道题的做法有很多，只是今天很快想到这种解法，若定义 n = len(haystack)，l = len(needle) 的话算法时间复杂度为O((n-l)*l)，我还是很菜…官方解法中有常数级时间复杂度的解法（思路是这样的，先生成窗口内子串的哈希码，然后再跟 needle 字符串的哈希码做比较），官方解法链接，有兴趣可以自查：Rabin Karp - 常数复杂度 0.0