跳至主要內容

剑指 Offer II 017. 含有所有字符的最短字符串

ITCharge大约 1 分钟

剑指 Offer II 017. 含有所有字符的最短字符串open in new window

  • 标签:哈希表、字符串、滑动窗口
  • 难度:困难

题目链接

题目大意

给定一个字符串 s、一个字符串 t

要求:返回 s 中涵盖 t 所有字符的最小子串。如果 s 中不存在涵盖 t 所有字符的子串,则返回空字符串 ""。如果存在多个符合条件的子字符串,返回任意一个。

解题思路

使用滑动窗口求解。

leftright 表示窗口的边界,一开始都位于下标 0 处。need 用于记录短字符串需要的字符数。window 记录当前窗口内的字符数。

right 右移,直到出现了 t 中全部字符,开始右移 left,减少滑动窗口的大小,并记录下最小覆盖子串的长度和起始位置。最后输出结果。

代码

class Solution:
    def minWindow(self, s: str, t: str) -> str:
        need = collections.defaultdict(int)
        window = collections.defaultdict(int)
        for ch in t:
            need[ch] += 1

        left, right = 0, 0
        valid = 0
        start = 0
        size = len(s) + 1

        while right < len(s):
            insert_ch = s[right]
            right += 1

            if insert_ch in need:
                window[insert_ch] += 1
                if window[insert_ch] == need[insert_ch]:
                    valid += 1

            while valid == len(need):
                if right - left < size:
                    start = left
                    size = right - left
                remove_ch = s[left]
                left += 1
                if remove_ch in need:
                    if window[remove_ch] == need[remove_ch]:
                        valid -= 1
                    window[remove_ch] -= 1
        if size == len(s) + 1:
            return ''
        return s[start:start + size]