赞
踩
Leetcode_28题目链接
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。
示例1:
输入:haystack = “sadbutsad”, needle = “sad”
输出:0
解释:“sad” 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。
参考-代码随想录
本题使用KMP算法解决,KMP算法是一种改进的字符串匹配算法,KMP的核心是利用匹配失败后的信息,尽量减少模式串与主串的匹配次数从而达到快速匹配的目的。
KMP算法的核心在于构建一个前缀表,前缀表的长度与模式串的长度一致,前缀表中每个数字的含义为以当前位置的字符作为结尾字符时,其之前的最长相等前后缀的长度;
前缀就是:不包含主串最后一个字符的以主串第一个字符开头的所有连续字串;
后缀就是:不包含主串第一个字符的以主串最后一个字符结尾的所有连续字串;
考虑模式串="aabaaf"和主串=“aabaabaabfa”
模式串子串 | 前缀 | 后缀 | 最长公共前后缀长度 | 模式串当前下标 |
---|---|---|---|---|
a | 无 | 无 | 0 | 0 |
aa | a | a | 1 | 1 |
aab | a,aa | b,ab | 0 | 2 |
aaba | a,aa,aab | a,ba,aba | 1 | 3 |
aabaa | a,aa,aab,aaba | a,aa,baa,abaa | 2 | 4 |
aabaaf | a,aa,aab,aaba,aabaa | f,af,aaf,baaf,abaaf | 0 | 5 |
可以看出模式串与前缀表对应位置的数字表示的是:下标i(包括i)的字符串中,有多大长度的相同前缀后缀
出现匹配失败的情况时,匹配失败的位置是在后缀子串的后面;此时通过前缀表记录的最长公共前后缀长度,可以知道最长相同的前缀和后缀,于是不用从头开始匹配模式串,而是回退到前缀后接着匹配就好了;
匹配失败后如何回到前缀后接着匹配?j=next[j-1]
生成前缀表即next
数组的过程与使用前缀表进行匹配的过程其实是一致的,即生成前缀表的过程就是模式串自己与自己匹配的过程;这里实际上是一个错位匹配,因为next
前缀表第一个元素一定是0,即其公共最长前后缀长度为0;那么就从下一位开始匹配,此时i指向后缀(以1起始),j指向前缀(以0起始)
设主串长度为m,模式串长度为n,匹配时间复杂度为O(n),生成前缀表的时间复杂度为O(m),KMP算法的时间复杂度是O(m+n);
使用KMP算法,构建前缀表即一个next
数组是必须的。
即构建一个next
数组,代码如下:
def getNext(s): # s为模式串 # 使用j指向模式串前缀末尾,使用i指向模式串后缀末尾 # 初始化next数组,注意到next数组第一个数字总是为0 j = 0 nextArr = [0] * len(s) nextArr[0] = j # i从1开始取值,遍历整个模式串 for i in range(1, len(mode_string)): # 考虑当前缀和后缀不相同的情况 # 肯定是当j>0时才开始比较 while j>0 and s[i] != s[j]: j = nextArr[j-1] # 考虑前后缀相同的情况 if s[i] == s[j]: j += 1 nextArr[i] = j return nextArr
代码如下:
class Solution: def strStr(self, haystack: str, needle: str) -> int: if len(needle) == 0: return 0 nextArr = self.construct(needle) print(nextArr) # 开始进行匹配 # j用于记录模式串起始位置 j = 0 # i用于记录主串的起使位置 for i in range(len(haystack)): # 不匹配的情况 # 回退到模式串前一个字符位置 while j >0 and haystack[i] != needle[j]: j = nextArr[j-1] # 匹配的情况 # 推进j if haystack[i] == needle[j]: j += 1 # 当j指向模式串的末尾,表示匹配成功,返回位置信息 if j == len(needle): return i - len(needle) + 1 # 不匹配返回-1 return -1 def construct(self, needle): nextArr = [0] * len(needle) j = 0 nextArr[0] = j for i in range(1, len(needle)): while j > 0 and needle[i] != needle[j]: j = nextArr[j-1] if needle[i] == needle[j]: j += 1 nextArr[i] = j return nextArr
2022/10/24:对构造前缀表的回退操作不是太清楚
2022/10/25:开摆
2022/10/29:没理解,写得乱…
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。