美文网首页@IT·互联网程序员面经集
面试难点!常用算法技巧之“滑动窗口”

面试难点!常用算法技巧之“滑动窗口”

作者: JAVA架构师的圈子 | 来源:发表于2021-04-26 16:52 被阅读0次

    算法简介

    滑动窗口,顾名思义,就是有一个大小可变的窗口,左右两端方向一致的向前滑动(右端固定,左端滑动;左端固定,右端滑动)。

    可以想象成队列,一端在push元素,另一端在pop元素,如下所示:

    假设有数组[a b c d e f g h]
    一个大小为3的滑动窗口在其上滑动,则有:

    [a b c]
      [b c d]
        [c d e]
          [d e f]
            [e f g]
              [f g h]
    

    适用范围

    • 1、一般是字符串或者列表

    • 2、一般是要求最值(最大长度,最短长度等等)或者子序列

    算法思想

    • 1、在序列中使用双指针中的左右指针技巧,初始化 left = right = 0,把索引闭区间 [left, right] 称为一个窗口。
    • 2、先不断地增加 right 指针扩大窗口 [left, right],直到窗口中的序列符合要求。
    • 3、此时,停止增加 right,转而不断增加 left 指针缩小窗口 [left, right],直到窗口中的序列不再符合要求。同时,每次增加 left前,都要更新一轮结果。
    • 4、重复第 2 和第 3 步,直到 right 到达序列的尽头。
      思路其实很简单:第 2 步相当于在寻找一个可行解,然后第 3 步在优化这个可行解,最终找到最优解。左右指针轮流前进,窗口大小增增减减,窗口不断向右滑动。

    算法模板

    1、单层循环

    def template():
        # 初始化滑动窗口两端
        left = right = 0
        
        # 序列及序列长度
        seq, seq_len = xx, xx
    
        # 滑动窗口序列
        slide_win = []
    
        # 结果值
        rst = xx
    
        while right < seq_len:
            slide_win.append(seq[right])
            # 还没找到一个可行解
            if not avaliable(slide_win):
                # 扩大窗口
                right += 1
            else:
                # 找到一个可行解,更新结果值
                rst = update()
                # 缩小窗口
                left += 1
    

    2、双层循环

    def template():
        # 初始化滑动窗口两端
        left = right = 0
        
        # 序列及序列长度
        seq, seq_len = xx, xx
    
        # 滑动窗口序列
        slide_win = []
    
        # 结果值
        rst = xx
    
        while right < seq_len:
            slide_win.append(seq[right])
            # 还没找到一个可行解
            if not avaliable(slide_win):
                # 扩大窗口
                right += 1
                continue
    
            # 循环更新可行解
            while avaliable(slide_win):
                # 找到一个可行解,更新结果值
                rst = update()
                # 缩小窗口
                left += 1
    

    模板只是一个解题思路,具体的题目可能需要具体分析,但是大体框架是不变的。
    记住: 多刷题,多总结,是王道

    算法示例

    1、最长不含重复字符的子字符串

    from collections import deque
    
    
    class Solution(object):
        def lengthOfLongestSubstring(self, s):
            """
            :type s: str
            :rtype: int
            """
            if not s:
                return 0
    
            index = 0
            
            # 因为这个滑动窗口需要从一端进,一端出,因此考虑采用队列
            slide_win = deque()
    
            # 因为在滑动过程中需要不断的从窗口中增减元素,因此需要一个变量来保持最大窗口长度
            max_len = 1
    
            while index < len(s):
                # 一个小的优化点:还没有遍历的元素长度加上当前的窗口长度已经小于最大窗口长度,则直接返回结果
                if len(slide_win) + (len(s) - index) <= max_len:
                    return max_len
    
                # 如果当前元素没有在滑动窗口中,则加入,并且窗口扩大
                if s[index] not in slide_win:
                    slide_win.append(s[index])
                    index += 1
                else:
                    # 如果当前元素已经在窗口中有值,则更新最大窗口长度
                    max_len = max(max_len, len(slide_win))
                    # 窗口缩小,对端不变
                    slide_win.popleft()
    
            return max(max_len, len(slide_win))
    

    2、绝对差不超过限制的最长连续子数组

    import heapq
    
    
    class Solution(object):
        def longestSubarray(self, nums, limit):
            """
            :type nums: List[int]
            :type limit: int
            :rtype: int
            """
            if not nums:
                return 0
    
            len_nums = len(nums)
    
            # 因为需要比较子列表中的最大差值,即需要知道子列表中的最大值和最小值
            # 因此采用大顶堆和小顶堆
            max_hq = []
            min_hq = []
    
            # 滑动窗口的前后索引
            left = 0
            right = 0
    
            # 在滑动窗口的过程中,更新最大的窗口长度
            max_win = 1
    
            while right < len_nums:
                if len_nums - left <= max_win:
                    return max_win
    
                # 将当前元素及其索引放入堆中,因为python只有小顶堆,因此这里采用负值来模拟大顶堆
                heapq.heappush(max_hq, (-nums[right], right))
                heapq.heappush(min_hq, (nums[right], right))
    
                # 窗口的最大差值
                diff = -max_hq[0][0] - min_hq[0][0]
    
                # 最大差值在允许范围内,窗口后边向前滑动,左边保持不变
                if diff <= limit:
                    right += 1
                    continue
    
                # 最大差值超过范围:
                # 更新最大窗口长度
                max_win = max(max_win, right - left)
    
                # 保证滑动窗口内的最大差值在允许范围内
                while -max_hq[0][0] - min_hq[0][0] > limit:
                    # 去掉大顶堆中在滑动窗口之外的所有最大元素
                    while max_hq[0][1] <= left:
                        heapq.heappop(max_hq)
                    # 去掉小顶堆中在滑动窗口之外的所有最小元素
                    while min_hq[0][1] <= left:
                        heapq.heappop(min_hq)
    
                    left += 1
    
            return max(max_win, right - left)
    

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

    from collections import deque
    class Solution(object):
        def lengthOfLongestSubstring(self, s):
            """
            :type s: str
            :rtype: int
            """
            if not s:
                return 0
            
            index = 0
    
            # 因为这里需要从窗口的两端进行元素的增加或减少,因此采用双端队列
            slide_win = deque()
    
            # 题目要求最长子串长度,因此需要在滑动过程中更新最大长度
            max_len = 0
    
            while index < len(s):
                ch = s[index]
    
                # 如果当前元素不在窗口中,则加入窗口
                if ch not in slide_win:
                    slide_win.append(ch)
                    # 窗口的右端向前滑动,左端不变(扩大窗口)
                    index += 1
                else:
                    # 当前元素已经存在窗口中,即表示找到一个可行解,更新最大长度
                    max_len = max(max_len, len(slide_win))
                      Java开发交流君样:756584822
                    # 将窗口中在当前元素值之前的元素全部移除掉,即窗口左端向前滑动,右端不变(缩小窗口)
                    while ch in slide_win:
                        slide_win.popleft()
            
            return max(max_len, len(slide_win))
    

    4、替换后的最长重复字符

    class Solution(object):
        def characterReplacement(self, s, k):
            """
            :type s: str
            :type k: int
            :rtype: int
            """
            if not s:
                return 0
            
            # 滑动窗口的左右两端
            left = 0
            right = 0
    
            # 记录每个字符出现的频率
            ch_fre = defaultdict(int)
    
            # 记录字符出现的最大频率
            max_fre = 0
    
            # 在窗口滑动过程中,更新最大长度
            max_len = 0
    
            while right < len(s):
                ch = s[right]
                
                # 当前字符频率加1
                ch_fre[ch] += 1
    
                # 更新字符出现过的最大频率
                max_fre = max(max_fre, ch_fre[ch])
    
                # 如果滑动窗口的长度大于字符能够达到的最大频率
                # 即窗口内不可能出现全是重复的字符,因此需要将窗口的左端向前滑动(缩小窗口)
                if right - left + 1 > max_fre + k:
                    # 滑动出去的字符需要将其频率减1
                    ch_fre[s[left]] -= 1
                    left += 1
    
                # 此时滑动窗口内全是重复字符,更新最大长度值 
                max_len = max(max_len, right - left + 1)
    
                # 滑动窗口的右端向前滑动(扩大窗口)
                right += 1
    //Java开发交流君样:756584822
            return max_len
    

    算法总结

    滑动窗口算法就是用以解决数组/字符串的子元素问题
    滑动窗口算法可以将嵌套的for循环问题,转换为单循环问题,降低时间复杂度

    生命不止坚毅鱼奋斗,有梦想才是有意义的追求
    给大家推荐一个免费的学习交流群:
    最后,祝大家早日学有所成,拿到满意offer,快速升职加薪,走上人生巅峰。
    Java开发交流君样:756584822

    相关文章

      网友评论

        本文标题:面试难点!常用算法技巧之“滑动窗口”

        本文链接:https://www.haomeiwen.com/subject/wyperltx.html