窗口最大值之和

👁️ 0 人浏览 💬 0 人评论 ❤️ 添加收藏

运行下面这段程序:

本节模型:window_max / window_min 用 std::deque 存下标做单调队列,求每个长度为 k 的窗口的最值。

#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({4, 2, 6, 1, 5}, 2);
    int s = 0;
    for (int x : r) s += x;
    cout << s << endl;
}

(本题用 g++ -std=c++17 -O0 编译。)

提交你的答案
请登录后提交答案。
去登录
代码编辑器
Ctrl + Enter 运行
本次输入:
输出:

                        
👩‍🏫
AI
💬 题目评论

全部评论