LeetCode：28. Implement strStr() - KMP算法 - Python
问题描述：
实现 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() 定义相符。
问题分析：
这个应该比较简单，KMP算法就可以解决了，当然Python也可以一行代码实现（return haystack.find(needle)），感觉这样就失去了这个题目的意义了哈。至于kmp这个经典算法，就不过多解释了，之所以总结是感觉以后这个算法还会用到。
Python3实现：
# @Time :2018/7/3
# @Author :LiuYinxing
# 解题思路 kmp 算法
class Solution:
def strStr(self, haystack, needle):
if len(needle) == 0: return 0
n, m = len(haystack), len(needle)
next = [0] * m
self.GetNext(needle, next) # 初始化next数组
i, j = 0, 0
while i < n and j < m:
if j == -1 or haystack[i] == needle[j]:
i, j = i + 1, j + 1
else:
j = next[j]
return i - m if j >= m else -1
def GetNext(self, t, next): # 计算next数组
j, k, next[0], n = 0, -1, -1, len(next)
while j < n - 1:
if k == -1 or t[j] == t[k]:
j, k = j + 1, k + 1
next[j] = k
else:
k = next[k]
if __name__ == '__main__':
haystack, needle = 'hello', 'll'
solu = Solution()
print(solu.strStr(haystack, needle))Python3改进版实现：
# @Time :2018/7/6
# @Author :LiuYinxing
# 解题思路 kmp 算法，next数组改进版
class Solution:
def strStr(self, haystack, needle):
if len(needle) == 0: return 0
n, m = len(haystack), len(needle)
next = [0] * m
self.GetNext(needle, next) # 初始化next数组
i, j = 0, 0
while i < n and j < m:
if j == -1 or haystack[i] == needle[j]:
i, j = i + 1, j + 1
else:
j = next[j]
return i - m if j >= m else -1
def GetNext(self, t, next): # 计算改进版next数组
j, k, next[0], n = 0, -1, -1, len(next)
while j < n - 1:
if k == -1 or t[j] == t[k]:
j, k = j + 1, k + 1
if t[j] != t[k]:
next[j] = k
else:
next[j] = next[k]
else:
k = next[k]
if __name__ == '__main__':
haystack, needle = 'hello', 'll'
solu = Solution()
print(solu.strStr(haystack, needle))欢迎指正哦