P5788 【模板】单调栈

题目：

P5788 【模板】单调栈 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

思路： 

定义单调递增栈 s用于存储数组元素下标（逆向存储），栈顶即为第一个比a[i]大的元素的下标（确保即将压入的当前元素为栈内最小，弹出栈顶比当前数小的）。

代码：

#include<iostream>
#include<cstdio>
#include<stack>
typedef unsigned long long ll;
const ll N = 10000010;
ll a[N], result[N];//a[N]是需要判断的数组（即输入的数组），result[N]是存储答案的数组
using namespace std;
stack<ll>s;//单调栈，用于存储下标
int main()
{
	ll n,i;
	cin >> n;
	for (i = 1; i <= n; i++)scanf("%lld", &a[i]);//输入数据
	for (i = n; i >= 1; i--) {//从后开始遍历
		while (!s.empty() && a[s.top()] < a[i])s.pop();//确保即将压入的当前元素为栈内最小，弹出栈顶比当前数小的
		result[i] = s.empty() ? 0 : s.top(); //存储答案，栈顶即为所求（若没有则输出0）
		s.push(i);//压入当前元素到栈顶
	}
	for (i = 1; i <= n; i++) printf("%lld ", result[i]); //输出
}