问题描述
约翰的N(1≤N≤100000)头奶牛站成一排,奶牛i的身高是Hi(l≤Hi≤1,000,000).现在,每只奶牛都在向右看齐.对于奶牛i,如果奶牛j满足i<j且Hi<Hj,我们可以说奶牛i可以仰望奶牛j.
求出每只奶牛离她最近的仰望对象.
输入格式
第1行输入N,之后每行输入一个身高.
输出格式
共N行,按顺序每行输出一只奶牛的最近仰望对象.如果没有仰望对象,输出0.
解析题目:
对于每只奶牛我们需要在它的右边找到第一个比它高的,然后输出(看起来好简单啊,哈哈哈哈哈)但是本题的难度来自于时间复杂度,所以肯定不能选择直接循环,我们要用到栈。
AC代码:
#include<bits/stdc++.h>
using namespace std;
int h[100005],ans[100005];
stack <int> cow;
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++) cin>>h[i];
for(int i=n;i>=1;i--){
while(!cow.empty()&&h[i]>=h[cow.top()])
cow.pop();
if(!cow.empty())
ans[i]=cow.top();
cow.push(i);
}
for(int i=1;i<=n;i++){
cout<<ans[i]<<endl;
}
return 0;
}