题目描述
一、思路
如果用大顶堆(优先队列)来做这道题,那就需要将窗口k个数字都放进大顶堆中,这样就可以知道窗口的最大值了,但是这个窗口是移动的,大顶堆每次都会把最大值给弹出,这样就无法将其他该弹出的数字弹出,这样就造成大顶堆维护的不是滑动窗口里面的数值了。
继续思考,我们需要一个队列,这个队列会在每次窗口移动的时候,调用que.pop(滑动窗口中移除元素的数值),que.push(滑动窗口添加元素的数值),然后que.front()就返回我们要的最大值。
但是队里的元素是一定要排序的,而且最大值要放到出队口,不然怎么才能知道最大值呢,那么问题来了,排序之后的队列 怎么能把窗口要移除的元素(这个元素不一定是最大值)弹出呢。
其实这个队列没必要维护窗口内的全部值,只需要维护窗口内有可能是最大值的元素即可,同时保证队列里的元素是由大到小的。那么这个维护元素单调递减的队列就叫做单调队列,即单调递减或单调递增的队列。
设计单调队列时pop和push操作要保持如下规则:
pop(val):如果窗口新入队的元素等于单调队列的出口元素,此时队列弹出元素,否则不进行任何操作;
push(val):如果新入队的元素大于入口元素的值,此时先将入口处的元素弹出,直到push元素的数值小于等于队列入口元素的数值为止;
保持如上规则,每次窗口移动的时候,只要问que.front()就可以返回当前窗口的最大值。
二、代码实现
1.单调队列的实现
代码如下(示例):
class MyQueue { //单调队列(从大到小)
public:
deque<int> que; // 使用deque来实现单调队列
// 每次弹出的时候,比较当前要弹出的数值是否等于队列出口元素的数值,如果相等则弹出。
// 同时pop之前判断队列当前是否为空。
void pop(int value) {
if (!que.empty() && value == que.front()) {
que.pop_front();
}
}
// 如果push的数值大于入口元素的数值,那么就将队列后端的数值弹出,直到push的数值小于等于队列入口元素的数值为止。
// 这样就保持了队列里的数值是单调从大到小的了。
void push(int value) {
while (!que.empty() && value > que.back()) {
que.pop_back();
}
que.push_back(value);
}
// 查询当前队列里的最大值 直接返回队列前端也就是front就可以了。
int front() {
return que.front();
}
};
2.整体代码
代码如下(示例):
class Solution {
private:
class MyQueue {
public:
deque<int> que;
void pop(int value) {
if (!que.empty() && value == que.front()) {
que.pop_front();
}
}
void push(int value) {
while (!que.empty() && value > que.back()) {
que.pop_back();
}
que.push_back(value);
}
int front() {
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> result;
for (int i = 0; i < k; i++) { // 先将前k的元素放进队列
que.push(nums[i]);
}
result.push_back(que.front()); // result 记录前k的元素的最大值
for (int i = k; i < nums.size(); i++) {
que.pop(nums[i - k]); // 滑动窗口移除最前面元素
que.push(nums[i]); // 滑动窗口前加入最后面的元素
result.push_back(que.front()); // 记录对应的最大值
}
return result;
}
};
总结
首先要明确的是,题解中单调队列里的pop和push接口,仅适用于本题。单调队列不是一成不变的,而是不同场景不同写法,总之要保证队列里单调递减或递增的原则,所以叫做单调队列。 不要以为本题中的单调队列实现就是固定的写法。
C++中deque是stack和queue默认的底层实现容器,deque是可以两边扩展的,而且deque里元素并不是严格的连续分布的。