今天就做了28题。暴力法和KMP都做了。暴力法很简单,KMP有点绕,尤其是里面要另外定义一个求next数组的函数。我对这种另外定义函数都有点犯怵。但是总体思路了解了,觉得并不是高不可攀的。现在可能有些细节还不清楚,但大体ok。
备注:没有完全看随想录视频。
class Solution:
def strStr(self, haystack: str, needle: str) -> int:
def build_next(): # 这一段是求next数组,其实可以先把下面的主体写出来,再写next数组。但是因为主体部分会call这个function,所以需要写在前面
#疑惑:def build_next(needle):括号里面要不要写needle?视频中是写的,还有下面主体部分也是写着的
next1 = [0] #初始化
prefix_len = 0 #目前,共同前缀长度为0
i = 1
while i < len(needle):
if needle[prefix_len] == needle[i]: #如果下一个字符依然相同,代表可以构成更长的的前后缀,长度+1
prefix_len += 1
next1.append(prefix_len)
i += 1
else: #如果下一个字符不同,那么直接查表,看看有没有更短的前后缀--跳到STEP A
if prefix_len == 0:
next1.append(0) #STEP B
i += 1
else:
prefix_len = next1[prefix_len - 1] #STEP A.如果还不存在,那么将next设为0(跳到STEP B)
return next1
next1 = build_next()#这里假设已经得到了next数组
#问题:next1 = build_next(needle)括号里面要不要写needle,不写也能pass
i = 0 #主串指针
j = 0 #子串指针
while i < len(haystack):
if haystack[i] == needle[j]: #如果子串和主串字符相同,则两边同时后移一位
i = i + 1
j = j + 1
elif j > 0:
j = next1[j-1] #匹配失败,根据next跳过子串前面的字符
else:
i = i + 1 #第一个字符就匹配失败
if j == len(needle):
return i - j
return -1