题目背景
模板题,无背景。
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;
}
标签:...,int,样例,a1,模板,include,单调
From: https://www.cnblogs.com/Xuxuan18/p/18564733