单调队列
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
本题数据量达到1e5,暴力的话显然超时,这里我们可以选用一种特殊结构单调对列
#include "bits/stdc++.h"
using namespace std;
class MonQueue{
public:
void push(int val) {
if(q.empty() || q.back() >= val) {
q.push_back(val);
}
else {
while(!q.empty() && q.back() < val) {
q.pop_back();
}
q.push_back(val);
}
}
void pop(int val) {
if(!q.empty() && q.front() == val) {
q.pop_front();
}
}
int front() {
return q.front();
}
private:
deque<int> q;
};
int main() {
int n, k;
cin>>n>>k;
vector<int> v(n);
for(int i = 0; i < n; i++) {
cin>>v[i];
}
vector<int> ans;
MonQueue q;
for(int i = 0; i < k; i++) {
q.push(v[i]);
}
ans.push_back(q.front());
for(int i = k; i < n; i++) {
q.push(v[i]);
q.pop(v[i - k]);
ans.push_back(q.front());
}
for(int i = 0; i < ans.size(); i++) {
cout<<ans[i]<<endl;
}
return 0;
}