【Leetcode】28. 找出字符串中第一个匹配项的下标

一、Problem

Leetcode_28题目链接
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标(下标从 0 开始)。如果 needle 不是 haystack 的一部分,则返回 -1 。

示例1:
输入:haystack = “sadbutsad”, needle = “sad”
输出:0
解释:“sad” 在下标 0 和 6 处匹配。
第一个匹配项的下标是 0 ,所以返回 0 。

二、Prerequisites

参考-代码随想录
本题使用KMP算法解决,KMP算法是一种改进的字符串匹配算法,KMP的核心是利用匹配失败后的信息,尽量减少模式串与主串的匹配次数从而达到快速匹配的目的。

  1. KMP算法的核心在于构建一个前缀表,前缀表的长度与模式串的长度一致,前缀表中每个数字的含义为以当前位置的字符作为结尾字符时,其之前的最长相等前后缀的长度

  2. 前缀就是:不包含主串最后一个字符的以主串第一个字符开头的所有连续字串;

  3. 后缀就是:不包含主串第一个字符的以主串最后一个字符结尾的所有连续字串;

  4. 考虑模式串="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
  5. 可以看出模式串与前缀表对应位置的数字表示的是:下标i(包括i)的字符串中,有多大长度的相同前缀后缀

  6. 出现匹配失败的情况时,匹配失败的位置是在后缀子串的后面;此时通过前缀表记录的最长公共前后缀长度,可以知道最长相同的前缀和后缀,于是不用从头开始匹配模式串,而是回退到前缀后接着匹配就好了;

  7. 匹配失败后如何回到前缀后接着匹配?j=next[j-1]

  8. 生成前缀表即next数组的过程与使用前缀表进行匹配的过程其实是一致的,即生成前缀表的过程就是模式串自己与自己匹配的过程;这里实际上是一个错位匹配,因为next前缀表第一个元素一定是0,即其公共最长前后缀长度为0;那么就从下一位开始匹配,此时i指向后缀(以1起始),j指向前缀(以0起始)
    【Leetcode】28. 找出字符串中第一个匹配项的下标_第1张图片

  9. 设主串长度为m,模式串长度为n,匹配时间复杂度为O(n),生成前缀表的时间复杂度为O(m),KMP算法的时间复杂度是O(m+n)

  10. 使用KMP算法,构建前缀表即一个next数组是必须的。

三、Code

3.1 构建前缀表

即构建一个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

3.2 利用前缀表和主串进行匹配

代码如下:

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:没理解,写得乱…

你可能感兴趣的:(Leetcode,1024程序员节)