单调队列经常配合滑动窗口,求滑动中动态的各窗口区间内元素最值类问题。
// 滑动窗口 + 单调队列(单调递减队列)
// 思路1:单调递减队列中存储数组元素下标
// 窗口滑动时,当前元素若大于(或大于等于)队尾下标对应的值则循环出队尾, 若当前队首下标在窗口外(队首下标小于等于left=i-k)则循环队首出队
class Solution {
public:
// 滑动窗口 + 单调队列(单调递减队列)
// 思路:
// 单调递增队列中存储数组下标, 下标是递增的, 下标对应的值是递减的
// 窗口滑动时,当前元素若大于(或大于等于)队尾下标对应的值则循环出队尾, 若当前队首下标在窗口外(队首下标小于等于left=i-k)则循环队首出队
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
if (nums.size() * k == 0) {
return {};
}
vector<int> res;
deque<int> q;
for (size_t i = 0; i < k; i++) {
while (!q.empty() && nums[q.back()] < nums[i]) {
q.pop_back();
}
q.push_back(i);
}
res.push_back(nums[q.front()]);
for (size_t i = k; i < nums.size(); i++) {
// 队列头是上一步窗口内最大值的索引值。若队列头不在当前窗口内,则队首出队. 循环比较
while (!q.empty() && q.front() <= i - k) {
q.pop_front();
}
while (!q.empty() && nums[q.back()] < nums[i]) {
q.pop_back();
}
q.push_back(i);
res.push_back(nums[q.front()]);
}
return res;
}
};
// 思路2:单调递减队列中存储数组元素值
// 实现单调递增队列
class MonotonicQueue {
public:
void push(int n) {
// 将小于 n 的元素全部删除
while (!q.empty() && q.back() < n) {
q.pop_back();
}
// 然后将 n 加入尾部
q.push_back(n);
}
int max() {
return q.front();
}
void pop(int n) {
if (n == q.front()) {
q.pop_front();
}
}
private:
deque<int> q;
};
class Solution1 {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
if (nums.size() * k == 0) {
return {};
}
MonotonicQueue window;
vector<int> res;
for (int i = 0; i < nums.size(); i++) {
if (i < k - 1) {
//先填满窗口的前 k - 1
window.push(nums[i]);
} else {
// 窗口向前滑动,加入新数字(在这里之前window中元素个数最多是k-1个,本步骤是加上新尾)
window.push(nums[i]);
// 记录当前窗口的最大值(上面一行加上新尾后,window中元素个数最多是k个)
res.push_back(window.max());
// 移出旧数字(去旧头)
window.pop(nums[i - k + 1]);
}
}
return res;
}
};