洛谷 P5788 【模板】單調棧

洛谷傳送門

題目背景

模板題，無背景。

2019.12.12 更新資料，放寬時限，現在不再卡常了。

題目描述

給出項數為 nn 的整數數列 a_{1 \dots n}a1…n。

定義函式 f(i)f(i) 代表數列中第 ii 個元素之後第一個大於 a_ia**i 的元素的下標，即 f(i)=\min_{i<j\leq n, a_j > a_i} {j}f(i)=mini<j≤n,a**j>a**i{j}。若不存在，則 f(i)=0f(i)=0。

試求出 f(1\dots n)f(1…n)。

輸入格式

第一行一個正整數 nn。

第二行 nn 個正整數 a_{1\dots n}a

輸出格式

一行 nn 個整數 f(1\dots 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;
}