数组模拟的窗口最大
运行下面这段程序:
本节模型:window_max_arr 不用 std::deque,用数组 q 加 head、tail 两个下标模拟两端操作。
#include <algorithm>
#include <deque>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
vector<int> window_max_arr(const vector<int>& a, int k) {
// 不用 std::deque:数组 q 加 head / tail 两个下标,q[head..tail-1] 就是队里的下标
int n = (int)a.size();
vector<int> q(n);
int head = 0, tail = 0;
vector<int> res;
for (int i = 0; i < n; i++) {
while (tail > head && a[q[tail - 1]] <= a[i]) tail--;
q[tail++] = i;
if (q[head] <= i - k) head++;
if (i >= k - 1) res.push_back(a[q[head]]);
}
return res;
}
string join(const vector<int>& v) {
string s;
for (size_t i = 0; i < v.size(); i++) s += (i ? "," : "") + to_string(v[i]);
return s;
}
int main() {
cout << join(window_max_arr({4, 2, 6, 1, 5}, 2)) << endl;
}
(本题用 g++ -std=c++17 -O0 编译。)
全部评论