几个有左边更大
运行下面这段程序:
本节模型:next_greater / prev_greater 用单调栈(vector 存下标)求右边 / 左边第一个更大的数,没有记 -1。
#include <algorithm>
#include <deque>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
vector<int> prev_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]) st.pop_back();
res[i] = st.empty() ? -1 : a[st.back()];
st.push_back(i);
}
return res;
}
int main() {
vector<int> r = prev_greater({1, 3, 2, 5});
cout << count_if(r.begin(), r.end(), [](int x) { return x != -1; }) << endl;
}
(本题用 g++ -std=c++17 -O0 编译。)
全部评论