【剑指Offer系列59】滑动窗口的最大值

题目

给定一个数组 nums 和滑动窗口的大小 k，请找出所有滑动窗口里的最大值。

代码

Python

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        deq=collections.deque() # 双端队列
        res,n=[],len(nums)
        for i,j in zip(range(1-k,n-k+1),range(n)): # 遍历窗口左右边界
            # 每轮窗口滑动移除了元素nums[i−1]，则将deq内的对应元素一起删除,以保持deq内仅包含窗口内的元素
            if i>0 and deq[0]==nums[i-1]: deq.popleft() 
            # 删除deq内所有小于nums[j]的元素，以保持deq递减
            while deq and deq[-1]<nums[j]: deq.pop()
            deq.append(nums[j])
            # 若已形成窗口，将窗口最大值添加至列表res
            if i>=0: res.append(deq[0])
        return res

C++

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> deq;
        vector<int> res;
        int n = nums.size();
        for (int i=0; i<n; i++) {
            while (!deq.empty() && nums[i]>nums[deq.back()]) {
                deq.pop_back();
            }   
            if (!deq.empty() && deq.front()<i-k+1) deq.pop_front();
            deq.push_back(i);
            if (i>=k-1) res.push_back(nums[deq.front()]);
        }
        return res;
    }
};