Python | Leetcode Python题解之第28题找出字符串中的第一个匹配项的下标

原创于 2024-04-14 08:27:54 发布 · 440 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Python #Leetcode #题解

分享专栏收录该内容

489 篇文章

订阅专栏

本文介绍了如何使用Python实现一个名为`strStr()`的函数，该函数利用偏移表优化字符串查找过程。它在haystack中查找needle的出现位置，展示了字符串处理和动态匹配策略。

题目：

题解：

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
    
        # Func: 计算偏移表
        def calShiftMat(st):
            dic = {}
            for i in range(len(st)-1,-1,-1):
                if not dic.get(st[i]):
                    dic[st[i]] = len(st)-i
            dic["ot"] = len(st)+1
            return dic
        
        # 其他情况判断
        if len(needle) > len(haystack):return -1
        if needle=="": return 0
       
        # 偏移表预处理    
        dic = calShiftMat(needle)
        idx = 0
    
        while idx+len(needle) <= len(haystack):
            
            # 待匹配字符串
            str_cut = haystack[idx:idx+len(needle)]
            
            # 判断是否匹配
            if str_cut==needle:
                return idx
            else:
                # 边界处理
                if idx+len(needle) >= len(haystack):
                    return -1
                # 不匹配情况下，根据下一个字符的偏移，移动idx
                cur_c = haystack[idx+len(needle)]
                if dic.get(cur_c):
                    idx += dic[cur_c]
                else:
                    idx += dic["ot"]
            
        
        return -1 if idx+len(needle) >= len(haystack) else idx