P5788 模板 【单调栈】

题目背景

模板题，无背景。

2019.12.12 更新数据，放宽时限，现在不再卡常了。
题目描述

给出项数为 nnn 的整数数列 a1…na_{1 \dots n}a1…n​。

定义函数 f(i)f(i)f(i) 代表数列中第 iii 个元素之后第一个大于 aia_iai​ 的元素的下标，即 f(i)=min⁡i<j≤n,aj>ai{j}f(i)=\min_{i<j\leq n, a_j > a_i} {j}f(i)=mini<j≤n,aj​>ai​​{j}。若不存在，则 f(i)=0f(i)=0f(i)=0。

试求出 f(1…n)f(1\dots n)f(1…n)。
输入格式

第一行一个正整数 nnn。

第二行 nnn 个正整数 a1…na_{1\dots n}a1…n​。
输出格式

一行 nnn 个整数 f(1…n)f(1\dots n)f(1…n) 的值。
输入输出样例
输入 #1

5
1 4 2 3 5

输出 #1

2 5 4 5 0

说明/提示

【数据规模与约定】

对于 30%30%30% 的数据，n≤100n\leq 100n≤100；

对于 60%60%60% 的数据，n≤5×103n\leq 5 \times 10^3n≤5×103 ；

对于 100%100%100% 的数据，1≤n≤3×1061 \le n\leq 3\times 10^61≤n≤3×106，1≤ai≤1091\leq a_i\leq 10^91≤ai​≤109。

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;

const int maxn = 3e6 + 10;
int a[maxn], st[maxn], ans[maxn];
int top;
int n;

int main() {
	scanf("%d", &n);
	for(int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	for(int i = n; i >= 1; i--) {
		while(top && a[st[top]] <= a[i])	top--;
		ans[i] = st[top];
		st[++top] = i;
//		cout << i << " " << st[top - 1] << " " << top << endl;
	} 
	for(int i = 1; i <= n; i++)	printf("%d ", ans[i]);
	return 0;
}