leetcode-239. 滑动窗口最大值

题目描述

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

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

示例 1：

输入：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

示例 2：

输入：nums = [1], k = 1
输出：[1]

思路

单调队列，参考：代码随想录

需自己构建一个队列，写清楚pop，push，front3个函数即可

1）push函数：队列要始终保持一个从大到小的列表，所以push操作的时候，需要把比当前值小的值，pop出去

2）pop函数：滑动窗口，需要逐次把最左边的数排出去；当时当前队列中只保留一部分从大到小的值【有些数在push操作中被去掉了】；所以当要排出去的数和队首的数一样时，则pop()

3）front函数：返回队首元素即可

from collections import deque

class Myqueue():
    def __init__(self):
        self.queue = deque()
    def push(self, x):
        while self.queue and x>self.queue[-1]: # 这里是-1，一点点去掉
            self.queue.pop()
        self.queue.append(x)
    def pop(self, x):
        if self.queue and x==self.queue[0]:
            self.queue.popleft()
    def front(self):
        return self.queue[0]

class Solution(object):
    def maxSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        que = Myqueue()
        result = []
        for i in range(k):
            que.push(nums[i])
        result.append(que.front())
        for j in range(k,len(nums)):
            que.pop(nums[j-k]) # 这里是j-k
            que.push(nums[j])
            result.append(que.front())
        return result

if __name__ == '__main__':
    s = Solution()
    nums = [1, 3, -1, -3, 5, 3, 6, 7]
    k = 3
    print(s.maxSlidingWindow(nums, k))