PS:如果读过题了可以跳过题目描述直接到题解部分
提交链接:洛谷 P5788 【模板】单调栈
题目
题目背景
模板题,无背景。
2019.12.12 更新数据,放宽时限,现在不再卡常了。
题目描述
给出项数为 n n n 的整数数列 a 1 … n a_{1 \dots n} a1…n。
定义函数 f ( i ) f(i) f(i) 代表数列中第 i i i 个元素之后第一个大于 a i a_i ai 的元素的下标,即 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(i)=0 f(i)=0。
试求出 f ( 1 … n ) f(1\dots n) f(1…n)。
输入格式
第一行一个正整数 n n n。
第二行 n n n 个正整数 a 1 … n a_{1\dots n} a1…n。
输出格式
一行 n n n 个整数 f ( 1 … n ) f(1\dots n) f(1…n) 的值。
样例 #1
样例输入 #1
5
1 4 2 3 5
样例输出 #1
2 5 4 5 0
提示
【数据规模与约定】
对于 30 % 30\% 30% 的数据, n ≤ 100 n\leq 100 n≤100;
对于 60 % 60\% 60% 的数据, n ≤ 5 × 1 0 3 n\leq 5 \times 10^3 n≤5×103 ;
对于 100 % 100\% 100% 的数据, 1 ≤ n ≤ 3 × 1 0 6 1 \le n\leq 3\times 10^6 1≤n≤3×106, 1 ≤ a i ≤ 1 0 9 1\leq a_i\leq 10^9 1≤ai≤109。
题解
首先判断栈顶的数是否小于准备加入栈的数,若小于,则记栈顶的数的对应的答案为准备加入的数并将其弹出,直到遇到第一个不小于的数,就把新的数加入栈,最后把所有还在栈里的数依次弹出并记他们的答案为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;
}