给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
class Solution:
def strStr(self, haystack: str, needle: str) -> int:
next = self.getNext(needle)
i1 = 0
i2 = 0
while i1<len(haystack) and i2<len(needle):
if haystack[i1] == needle[i2]:
i1+=1
i2+=1
else:
if next[i2]==-1:
i1+=1
else:
i2=next[i2]
if i2==len(needle):
return i1-i2
else:
return -1
def getNext(self, needle):
if len(needle)<2:
return [-1]
next = [-1] * len(needle)
next[0] = -1
next[1] = 0
// cn是当前位置的最大前缀最后一个字符index+1
// i是当前位置,i的最大前后缀要比较i-1
cn = 0
i = 2
while i<len(needle):
if needle[i-1] == needle[cn]:
next[i] = cn+1
cn += 1
i += 1
else:
if cn>0:
cn = next[cn]
else:
next[i] = 0
i+=1
return next