【Leetcode笔记】找出字符串中第一个匹配项的下标_leetcode找出字符串第一个匹配的下标-CSDN博客

本文链接：https://blog.csdn.net/zyzcuczyu/article/details/129283875

Leetcode原题链接：找出字符串中第一个匹配项的下标

一、暴力版

1、思路

简单的遍历法。
把不满足的条件罗列出，剩下的就是能满足的。

2、代码

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        n_p = 0
        haystack, needle = list(haystack), list(needle)
        ret = -1
        count = 0

        for h_p in range(len(haystack)):
            if len(haystack) - h_p < len(needle):
                return -1
            else:
                if haystack[h_p] != needle[n_p]:
                    continue
                else:
                    for i in range(len(needle)):
                        if haystack[h_p + i] == needle[i]:
                            count += 1
                    if count != len(needle):
                        count = 0
                        continue
                    else:
                        ret = h_p
                        return ret      
        return ret

二、KMP算法

1、思路

1.1 给模式串做一个`next`列表，用来保存当前位置子串的最大前后缀相同的长度

①初始化next列表（next[0] = 0，毕竟单个字符串没有前后缀）和j（当前子串前缀的最后一个字符，即j = 0），i（当前子串后缀的最后一个字符）无需初始化，下面的循环将从i = 1开始。
②遍历模式串的子串，若当前子串的i、j位置的字符不同，则j退回到当前字符前一个字符对应的next列表值的位置，即j = next[j-1]（如果一直不同，则一直退回到模式串第一个字符的位置，即j = 0处），赋值给当前i位置的next列表值为j，即next[i] = j。【注意此处是while循环】
③最终得到长度为模式串长度的next列表。

1.2 借助`next`列表对文本串进行逐个字符比较

①若当前字符不同，则j退回到当前字符前一个字符对应的next列表值的位置，即j = next[j-1]（如果一直不相同，则一直退回到模式串第一个字符的位置，即j = 0），而后继续循环i。【注意此处是while循环】
②若当前字符相同，则i、j分别后移继续比对。直到j等于模式串的长度，返回i-j+1；或是匹配失败，退出循环，返回-1。

2、代码

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        if len(haystack) < len(needle) or len(needle) == 0:
            return -1
        
        next = self.getNext(needle)
        j = 0
        for i in range(len(haystack)):
            while(j > 0 and haystack[i] != needle[j]):
                j = next[j-1]
            if haystack[i] == needle[j]:
                j += 1
            if j == len(needle):
                return i-j+1
        return -1
    
    def getNext(self, needle):
        next = [0] * len(needle)
        j = 0
        next[0] = 0
        for i in range(1, len(needle)):
            while(j > 0 and needle[j] != needle[i]):
                j = next[j-1]
            if needle[j] == needle[i]:
                j += 1
            next[i] = j
        return next