【模板】单调栈

题目背景

模板题，无背景。

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

题目描述

给出项数为 n 的整数数列 a1...n。

定义函数 f(i) 代表数列中第 i 个元素之后第一个大于 ai 的元素的下标，即
。若不存在，则 f(i)=0。

试求出 f(1...n)。

输入格式

第一行一个正整数 n。

第二行 n 个正整数 a1...n。

输出格式

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

样例 #1

样例输入 #1

5
1 4 2 3 5

样例输出 #1

2 5 4 5 0

提示

【数据规模与约定】

对于 30% 的数据，n<=100；

对于 60% 的数据，n<=10^3 ；

对于 100% 的数据，1 <= n<= 10^6，1<=n<= 10^9。

代码示例

#include<iostream>
#include<stack>
using namespace std;
int n,a[3000005],f[3000005];
stack<int>s;//模拟用的栈
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=n;i>=1;i--)
	{
		while(!s.empty()&&a[s.top()]<=a[i]) s.pop();//弹出栈顶比当前数小的
		f[i]=s.empty()?0:s.top();//存储答案
		s.push(i);//压入当前元素
	}
	for(int i=1;i<=n;i++) printf("%d ",f[i]);
	return 0;
}