LeetCode 239. 滑动窗口最大值

题目描述

239. 滑动窗口最大值

解法一：优先队列（C++）

对于「最大值」，我们可以想到一种非常合适的数据结构，那就是优先队列（堆），其中的大根堆可以帮助我们实时维护一系列元素中的最大值

而在这里，我们还需要记录每个元素的索引，因为当最大值的索引小于等于 $i-k$ 时，其实最大元素已经落到了滑窗的左侧，此时需要将其从堆中移除

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        priority_queue<pair<int, int>> q;
        vector<int> res;

        for(int i=0;i<k;i++)
            q.emplace(nums[i], i);
        res.push_back(q.top().first);
        for(int i=k;i<n;i++)
        {
            q.emplace(nums[i], i);
            while(q.top().second<=i-k) q.pop();
            res.push_back(q.top().first);
        }
        return res;
    }
};

解法二：单调队列

我们先解释一下什么叫做单调队列？

首先，我们希望每次窗口移动的时候，调用 pop 就可以在队列中移除不再落入滑窗的值，调用 push 在队列中插入窗口中心添加的元素，队列中元素成递减排列，那么调用 front 就可以得到滑窗的最大值，这种数据结构就称为单调队列

但是没有这种现成的数据结构，需要我们自己实现

---

设计单调队列时，底层采用的还是 STL 提供的 deque

pop(value) ：如果窗口移除的元素（value）等于单调队列的出口元素，那么队列弹出元素，否则不用任何操作

push(vaule) ：如果新添加的元素（value）大于入口元素的数值，那么就将队列入口的元素弹出，直到新添加元素的数值小于等于队列入口元素的数值为止

---

看个例子，

C++ 实现

class Solution {

private:
    class MonotonicQueue{
    public:
        deque<int> q;

        void pop(int value){
            if(!q.empty()&&value==q.front()) q.pop_front();
        }

        void push(int value){
            while(!q.empty()&&value>q.back())
                q.pop_back();
            q.push_back(value);
        }

        int front(){
            return q.front();
        }

    };

public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MonotonicQueue mq;
        vector<int> res;
        for(int i=0;i<k;i++)
            mq.push(nums[i]);
        res.push_back(mq.front());
        for(int i=k;i<nums.size();i++)
        {
            mq.pop(nums[i-k]);
            mq.push(nums[i]);
            res.push_back(mq.front());
        }
        return res;
    }
};

Python 实现

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        deque = collections.deque()
        res = []
        for i, num in enumerate(nums):
            while deque and deque[0] <= i-k: deque.popleft()
            while deque and num > nums[deque[-1]]: deque.pop()
            deque.append(i)
            if i >= k - 1:
                res.append(nums[deque[0]])
        return res