综合的窗口最大值之和
运行下面这段程序:
综合:按题目选单调栈(最近更值)还是单调队列(滑窗最值)。
#include <algorithm>
#include <deque>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
vector<int> window_max(const vector<int>& a, int k) {
// 每个长度为 k 的窗口的最大值
deque<int> dq; // 存下标,对应的值从队首到队尾递减
vector<int> res;
for (int i = 0; i < (int)a.size(); i++) {
while (!dq.empty() && a[dq.back()] <= a[i]) dq.pop_back();
dq.push_back(i);
if (dq.front() <= i - k) dq.pop_front();
if (i >= k - 1) res.push_back(a[dq.front()]);
}
return res;
}
int main() {
vector<int> r = window_max({3, 1, 4, 1, 5}, 2);
int s = 0;
for (int x : r) s += x;
cout << s << endl;
}
(本题用 g++ -std=c++17 -O0 编译。)
全部评论