滑动窗口算法

在滑动窗口类型的问题中都会有两个指针,一个用于延伸现有窗口的 right 指针,和一个用于收缩窗口的 left 指针。在任意时刻,只有一个指针运动,而另一个保持静止。滑动窗口算法的时间复杂度为 O(n),因为 rightleft 两个指针都只移动了 n 次。
该算法的大致逻辑如下:

int left = 0, right = 0;

while (right < s.size()) {
    // 增大窗口
    window.add(s[right]);
    right++;
    
    while (window needs shrink) {
        // 缩小窗口
        window.remove(s[left]);
        left++;
    }
}

接下来还是看几道典型的 leetcode 题目。

209. 长度最小的子数组

题目描述:

给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度
最小的连续子数组。如果不存在符合条件的连续子数组,返回 0。 

示例: 
输入: s = 7, nums = [2,3,1,2,4,3]
输出: 2
解释: 子数组 [4,3] 是该条件下的长度最小的连续子数组。

进阶: 
如果你已经完成了O(n) 时间复杂度的解法, 请尝试 O(nlogn) 时间复杂度的解法。 
Related Topics 数组 双指针 二分查找

题目分析:
首先想到暴力解法,循环每个子区间,找到满足和大于等于 s 的子区间,更新最小长度。

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        if nums is None or len(nums) == 0:
            return 0
        n = len(nums)
        res = n + 1
        for i in range(n):
            numSums = 0
            for j in range(i, n):
                numSums += nums[j]
                if numSums >= s:
                    res = min(res, j-i+1)
                    break
        return 0 if res == n+1 else res

此时时间复杂度 O(n^2),题目提示复杂度应为 O(n),那需要在常数次遍历中找到答案。在暴力解法中,我们先固定 i,然后增加 j,一次结束后,我们将 i1,然后再循环 j,重复的计算主要在这里。那是不是可以保持 j 在上一次循环结束的地方不动呢?我们知道上一次循环结束后 ij 之间的子数组的和大于等于 s,如果此时 j 不动,然后 i1,那我们求得的和就需要减掉 nums[i],如果此时的子数组的和满足条件,那最小长度比开始更小了,那我们需要更新最小长度;如果此时子数组的和不满足条件,那我们继续增加 j,来扩大子数组的长度。这也就是我们的滑动窗口思路。

class Solution:
    def minSubArrayLen(self, s: int, nums: List[int]) -> int:
        if nums is None or len(nums) == 0:
            return 0
        left, right, numSum = 0, 0, 0
        res = len(nums) + 1
        while right < len(nums):
            # 扩大窗口
            numSum += nums[right]
            while numSum >= s:
                res = min(res, right-left+1)
                # 缩小窗口
                numSum -= nums[left]
                left += 1
            right += 1
        return 0 if res == len(nums) + 1 else res

76. 最小覆盖子串

题目描述:

给你一个字符串 S、一个字符串 T,请在字符串 S 里面找出:包含 T 所有字母的最小子串。 

示例: 
输入: S = "ADOBECODEBANC", T = "ABC"
输出: "BANC" 
 
说明: 
如果 S 中不存这样的子串,则返回空字符串 ""。 
如果 S 中存在这样的子串,我们保证它是唯一的答案。 

Related Topics 哈希表 双指针 字符串 Sliding Window

联系滑动窗口算法,首先我们增大窗口,当条件满足时,缩小窗口。考虑到 t中可能有重复字符,所以我们的判断条件还需要考虑每个字符的数量一致。

class Solution:
    def minWindow(self, s: str, t: str) -> str:
        if s is None or t is None:
            return ''
        left, right, valid = 0, 0, 0
        start, length = 0, len(s)+1
        need, windown = {}, {}
        for i in t:
            need.setdefault(i, 0)
            need[i] += 1
        while right < len(s):
            # 增大窗口
            if s[right] in need:
                windown.setdefault(s[right], 0)
                windown[s[right]] += 1
                if need[s[right]] == windown[s[right]]:
                    valid += 1
            right += 1
            # 条件满足
            while valid == len(need):
                if right - left < length:
                    length = min(length, right-left)
                    start = left
                # 缩减窗口
                if s[left] in need:
                    if windown[s[left]] == need[s[left]]:
                        valid -= 1
                    windown[s[left]] -= 1
                left += 1
        return '' if length == len(s)+1 else s[start: start+length+1]

438. 找到字符串中所有字母异位词

题目描述:

给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,
返回这些子串的起始索引。 

字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。 
说明: 
字母异位词指字母相同,但排列不同的字符串。 
不考虑答案输出的顺序。 

示例 1: 
输入:
s: "cbaebabacd" p: "abc"
输出:
[0, 6]
解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的字母异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的字母异位词。

示例 2: 
输入:
s: "abab" p: "ab"
输出:
[0, 1, 2]
解释:
起始索引等于 0 的子串是 "ab", 它是 "ab" 的字母异位词。
起始索引等于 1 的子串是 "ba", 它是 "ab" 的字母异位词。
起始索引等于 2 的子串是 "ab", 它是 "ab" 的字母异位词。

Related Topics 哈希表

题目分析:
这道题需要关注的窗口为固定长度。我们先不考虑这个点,看看普通的滑动窗口解法:

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        need, windown = {}, {}
        res = []
        left, right, valid = 0, 0, 0
        for a in p:
            need.setdefault(a, 0)
            need[a] += 1
        while right < len(s):
            b = s[right]
            right += 1
            if b in need:
                windown.setdefault(b, 0)
                windown[b] += 1
                if need[b] == windown[b]:
                    valid += 1
            while valid == len(need):
                if right - left == len(p):
                    res.append(left)
                c = s[left]
                if c in windown:
                    if windown[c] == need[c]:
                        valid -= 1
                    windown[c] -= 1
                left +=1

        return res

这个解法中,我们是等到子字符串满足条件了再来缩减窗口。由于题目所述为固定的窗口大小,那我们缩减窗口的条件可以改成如下形式:

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        need, windown = {}, {}
        res = []
        left, right, valid = 0, 0, 0
        for a in p:
            need.setdefault(a, 0)
            need[a] += 1
        while right < len(s):
            b = s[right]
            right += 1
            if b in need:
                windown.setdefault(b, 0)
                windown[b] += 1
                if need[b] == windown[b]:
                    valid += 1
            # 这里用 if 判断亦可
            while right - left >= len(p):
                if valid == len(need):
                    res.append(left)
                c = s[left]
                left +=1 
                if c in need:
                    # 这个判断条件很重要,只有当相等的时候才需要将 valid 减 1
                    if windown[c] == need[c]:
                        valid -= 1
                    windown[c] -= 1
        return res

567. 字符串的排列

题目描述:

给定两个字符串 s1 和 s2,写一个函数来判断 s2 是否包含 s1 的排列。 
换句话说,第一个字符串的排列之一是第二个字符串的子串。 

示例1: 
输入: s1 = "ab" s2 = "eidbaooo"
输出: True
解释: s2 包含 s1 的排列之一 ("ba").

示例2: 
输入: s1= "ab" s2 = "eidboaoo"
输出: False

注意: 
输入的字符串只包含小写字母 
两个字符串的长度都在 [1, 10,000] 之间 

Related Topics 双指针 Sliding Window

和上一题类似,我们还是套用滑动窗口模板:

class Solution:
    def checkInclusion(self, s1: str, s2: str) -> bool:
        need, windown = {}, {}
        for a in s1:
            need.setdefault(a, 0)
            need[a] += 1
        left, right, valid = 0, 0, 0
        while right < len(s2):
            b = s2[right]
            right += 1
            if b in need:
                windown.setdefault(b, 0)
                windown[b] += 1
                if windown[b] == need[b]:
                    valid += 1
            # 缩小窗口
            if right - left >= len(s1):
                # 判断是否满足题目要求
                if valid == len(need):
                    return True
                c = s2[left]
                left += 1
                if c in need:
                    if windown[c] == need[c]:
                        valid -= 1
                    windown[c] -= 1
        return False

3. 无重复字符的最长子串

题目描述:

给定一个字符串,请你找出其中不含有重复字符的 最长子串 的长度。 

示例 1: 
输入: "abcabcbb"
输出: 3 
解释: 因为无重复字符的最长子串是 "abc",所以其长度为 3。

示例 2: 
输入: "bbbbb"
输出: 1
解释: 因为无重复字符的最长子串是 "b",所以其长度为 1。

示例 3: 
输入: "pwwkew"
输出: 3
解释: 因为无重复字符的最长子串是 "wke",所以其长度为 3。
请注意,你的答案必须是 子串 的长度,"pwke" 是一个子序列,不是子串。

Related Topics 哈希表 双指针 字符串 Sliding Window

题目分析:
题目的条件是无重复字符,那我们滑动窗口的条件便是:没有重复字符时,我们增大窗口,出现重复字符时我们缩小滑动窗口。需要注意的是我们更新结果再增加窗口的步骤中。

class Solution:
    def lengthOfLongestSubstring(self, s: str) -> int:
        # 无重复字符,我们增加窗口;有重复字符时,我们缩小窗口
        windown = set()
        left, right, valid = 0, 0, 0
        res = 0
        while right < len(s):
            a = s[right]
            right += 1
            # 有重复字符,需要缩减窗口
            while a in windown:
                b = s[left]
                left += 1
                windown.remove(b)
            else:
                windown.add(a)
                res = max(right-left, res)
        return res

4. 340. 至多包含 K 个不同字符的最长子串

题目描述

给你一个字符串 s 和一个整数 k ,请你找出 至多 包含 k 个 不同 字符的最长子串,并返回该子串的长度。

示例 1:
输入:s = "eceba", k = 2
输出:3
解释:满足题目要求的子串是 "ece" ,长度为 3 。
示例 2:
输入:s = "aa", k = 1
输出:2
解释:满足题目要求的子串是 "aa" ,长度为 2 。
 
提示:
1 <= s.length <= 5 * 104
0 <= k <= 50

题目分析:
还是采用滑动窗口算法,直接看解答吧。

import java.util.HashMap;

public class SlidingWindow {

    public static void main(String[] args) {

        // 给你一个字符串 s 和一个整数 k ,请你找出 至多 包含 k 个 不同 字符的最长子串,并返回该子串的长度。
        String s = "aa";
        int length = maxLength(s, 1);
        System.out.println(length);
    }

    public static int maxLength(String s, int k) {
        HashMap<Character, Integer> window = new HashMap<>();
        int left  = 0, right = 0;
        int res = 0;
        while (right < s.length()) {
            char c = s.charAt(right);
            right++;
            while (window.containsKey(c) && window.size() > k) {
                char d = s.charAt(left);
                left++;
                window.put(d, window.getOrDefault(d, 0)-1);
                if (window.get(d) == 0) {
                    window.remove(d);
                }
            }
            window.put(c, window.getOrDefault(c, 0)+1);
            if (window.size() == k) {
                res = Math.max(res, right-left);
            }
        }
        return res;
    }
}

参考

  1. https://leetcode-cn.com/problems/minimum-window-substring/solution/hua-dong-chuang-kou-suan-fa-tong-yong-si-xiang-by-/
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,367评论 6 512
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,959评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,750评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,226评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,252评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,975评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,592评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,497评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,027评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,147评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,274评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,953评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,623评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,143评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,260评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,607评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,271评论 2 358