洛谷 P5788 【模板】单调栈

PS：如果读过题了可以跳过题目描述直接到题解部分
提交链接：洛谷 P5788 【模板】单调栈

题目

题目背景

模板题，无背景。

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

题目描述

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

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

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

输入格式

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

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

输出格式

一行 $n$ 个整数 $f(1\dots 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$。

题解

首先判断栈顶的数是否小于准备加入栈的数，若小于，则记栈顶的数的对应的答案为准备加入的数并将其弹出，直到遇到第一个不小于的数，就把新的数加入栈，最后把所有还在栈里的数依次弹出并记他们的答案为0。最后按顺序输出答案即可。

代码实现

100pts

//洛谷 P5788 【模板】单调栈
#include<iostream>
#include<cstdio>
using namespace std;
int n;
int a[3000010];
int b[3000010];
int ans[3000010];

int main(){
	register int i,j;
	cin>>n;
	for(i=1;i<=n;++i){
		cin>>a[i];
	}
	for(i=1,j=0;i<=n;++i){
		while(j&&a[b[j]]<a[i]){
			ans[b[j]]=i;
			--j;
		}
		b[++j]=i;
	}
	for(;j>0;--j){
		ans[b[j]]=0;
	}
	for(i=1;i<=n;++i){
		cout<<ans[i]<<' ';
	}
	return 0;
}