【模板】单调栈

【模板】单调栈

题目描述

给出项数为 $n$ 的整数数列 $a_{1\dots n}$。

定义函数 $f(i)$ 代表数列中第 $i$ 个元素之后第一个大于 $a_i$ 的元素的下标，即 $f(i)={\min }_{i<j\le n,a_j>a_i}\{j\}$。若不存在，则 $f(i)=0$。

试求出 $f(1\dots n)$。

输入格式

第一行一个正整数 $n$。

第二行 $n$ 个正整数 $a_{1\dots n}$。

输出格式

一行 $n$ 个整数表示 $f(1),f(2),\dots ,f(n)$ 的值。

样例 #1

样例输入 #1

5
1 4 2 3 5

样例输出 #1

2 5 4 5 0

提示

【数据规模与约定】

对于 $30\%$ 的数据，$n\le 100$；

对于 $60\%$ 的数据，$n\le 5\times 10^3$ ；

对于 $100\%$ 的数据，$1\le n\le 3\times 10^6$，$1\le a_i\le 10^9$。

#include
using namespace std;
//#define int long long
const int N=1e5+5;
int n,a[N],cnt,x;
void solve(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&x);
		while(cnt>0&&a[cnt]>=x) cnt--;
		if(cnt==0) printf("-1 ");
		else printf("%d ",a[cnt]);
		a[++cnt]=x;
	}
	puts("");
}
signed main(){
	int t=1;
//	scanf("%d",&t);
	while(t--) solve();
	return 0;
}