【力扣】239:滑动窗口最大值 | 维护单调队列

题目描述

给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值。

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7] 
解释: 

  滑动窗口的位置                最大值
---------------               -----
[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7

算法思路

最简单的方法,维持滑动窗口的单调队列。

每滑动一下,窗口就丢掉一个值,加入一个值,就需要维护队列。

        def add(n):
            while True:
                if not ls or ls[-1]>=n:
                    ls.append(n)
                    break
                else:
                    ls.pop()
        def delete(n):
            if ls and ls[0]==n:
                    ls.pop(0)

完整算法

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if not nums:return []
        
        def add(n):
            while True:
                if not ls or ls[-1]>=n:
                    ls.append(n)
                    break
                else:
                    ls.pop()
        def dep(n):
            if ls and ls[0]==n:
                    ls.pop(0)
                
        ls=[]
        for i in range(k):
            add(nums[i])
        res=[]
        for i in range(k-1,len(nums)):
            res.append(ls[0])
            dep(nums[i+1-k])
            if i+1<len(nums):
                add(nums[i+1])
        return res

执行用时 :88 ms, 在所有 Python3 提交中击败了65.88%的用户
内存消耗 :17.1 MB, 在所有 Python3 提交中击败了100.00%的用户

你可能感兴趣的:(力扣日记)