洛谷 P5788 【模板】单调栈

洛谷 P5788 【模板】单调栈

洛谷传送门

题目背景

模板题，无背景。

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

题目描述

给出项数为 nn 的整数数列 a_{1 dots n}a1…n。

定义函数 f(i)f(i) 代表数列中第 ii 个元素之后第一个大于 a_ia**i 的元素的下标，即 f(i)=min_{i<jleq n, a_j > a_i} {j}f(i)=mini<j≤n,a**j>a**i{j}。若不存在，则 f(i)=0f(i)=0。

试求出 f(1dots n)f(1…n)。

输入格式

第一行一个正整数 nn。

第二行 nn 个正整数 a_{1dots n}a1…n。

输出格式

一行 nn 个整数 f(1dots n)f(1…n) 的值。

---

题解：

单调栈模板题。

代码：

#include<cstdio>
using namespace std;
const int maxn=3e6+9;
int n;
int a[maxn],ans[maxn];
int s[maxn],top;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	a[n+1]=2e9;
	for(int i=1;i<=n+1;i++)
	{
		while(top && a[i]>a[s[top]])
		{
			ans[s[top]]=i;
			top--;
		}
		s[++top]=i;
	}
	for(int i=1;i<=n;i++)
	{
		if(ans[i]==n+1)
			ans[i]=0;
		printf("%d ",ans[i]);
	}
	return 0;
}