几个元素有更大的

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

运行下面这段程序:

本节模型:next_greater / prev_greater 用单调栈(vector 存下标)求右边 / 左边第一个更大的数,没有记 -1。

#include <algorithm>
#include <deque>
#include <iostream>
#include <string>
#include <vector>
using namespace std;

vector<int> next_greater(const vector<int>& a) {
    // 每个元素右边第一个更大的数,没有记 -1
    int n = (int)a.size();
    vector<int> res(n, -1);
    vector<int> st;   // 单调栈:存「还没等到更大数」的下标
    for (int i = 0; i < n; i++) {
        while (!st.empty() && a[st.back()] < a[i]) {
            res[st.back()] = a[i];
            st.pop_back();
        }
        st.push_back(i);
    }
    return res;
}

int main() {
    vector<int> r = next_greater({1, 3, 2, 4});
    cout << count_if(r.begin(), r.end(), [](int x) { return x != -1; }) << endl;
}

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

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

                        
👩‍🏫
AI
💬 题目评论

全部评论