题目描述
给出项数为 nn 的整数数列 a_{1 \dots n}a1…n。
定义函数 f(i)f(i) 代表数列中第 ii 个元素之后第一个大于 a_iai 的元素的下标,即 f(i)=\min_{i<j\leq n, a_j > a_i} \{j\}f(i)=mini<j≤n,aj>ai{j}。若不存在,则 f(i)=0f(i)=0。
试求出 f(1\dots n)f(1…n)。
输入格式
第一行一个正整数 nn。
第二行 nn 个正整数 a_{1\dots n}a1…n。
输出格式
一行 nn 个整数 f(1\dots n)f(1…n) 的值。
输入输出样例
输入 #1复制
5 1 4 2 3 5
输出 #1复制
2 5 4 5 0
说明/提示
【数据规模与约定】
对于 30\%30% 的数据,n\leq 100n≤100;
对于 60\%60% 的数据,n\leq 5 \times 10^3n≤5×103 ;
对于 100\%100% 的数据,1 \le n\leq 3\times 10^61≤n≤3×106,1\leq a_i\leq 10^91≤ai≤109。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e6+5;
ll n,a[maxn],f[maxn];
stack<ll>s;
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n;
for(ll i=1;i<=n;i++)cin>>a[i];
//倒序循环
for(ll 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++)
{
cout<<f[i]<<" ";
}
return 0;
}