洛谷单调栈板子题

题目

P5788 【模板】单调栈

思路

用数组当作单调栈进行模拟，操作简单，速度快。

AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 3e6 + 10;
int stk[N], tt, a[N];

int main() {
	int n;
	cin.tie(0);//数据量过大加优化才能过
	ios::sync_with_stdio(false);
	cin >> n;
	for (int i = 0; i < n; ++i) {
		cin >> a[i];
	}
	vector<int> res;
	for (int i = n - 1; i >= 0; --i) {
		if (i == n - 1) {
			res.emplace_back(0);
			stk[++tt] = i;
			continue;
		}
		while (tt and a[stk[tt]] <= a[i])
			tt--;
		if (tt)
			res.emplace_back(stk[tt] + 1);
		else
			res.emplace_back(0);
		stk[++tt] = i;//入栈
	}
	int len = res.size();
	for (int i = len - 1; i >= 0; --i) {
		cout << res[i] << " ";
	}
}