2024华为OD机试题库-(C卷+D卷)-(JAVA、Python、C++)
题目描述
在学校中,N个小朋友站成一队, 第i个小朋友的身高为height[i],
第i个小朋友可以看到的第一个比自己身高更高的小朋友j,那么j是i的好朋友(要求j > i)。
请重新生成一个列表,对应位置的输出是每个小朋友的好朋友位置,如果没有看到好朋友,请在该位置用0代替。
小朋友人数范围是 [0, 40000]。
输入描述
第一行输入N,N表示有N个小朋友
第二行输入N个小朋友的身高height[i],都是整数
输出描述
输出N个小朋友的好朋友的位置
用例1
输入
2
100 95
输出
0 0
说明
第一个小朋友身高100,站在队尾位置,向队首看,没有比他身高高的小朋友,所以输出第一个值为0。
第二个小朋友站在队首,前面也没有比他身高高的小朋友,所以输出第二个值为0。
用例2
输入
6
23 56 45 18 100 95
输出
1 4 4 4 0 0
用例3
输入
6
23 56 45 48 100 95
输出
1 4 3 4 0 0
解题思路
1.栈为空时,先让元素下标入栈,新的元素下标与栈顶元素下标比较
2.如果新元素大于栈顶元素值,则更新栈顶元素下标的结果数组
考点
单调栈
代码
c++
#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
cin>>n;
vector<int> vec(n);
for(int i=0;i<n;i++) {
cin>>vec[i];
}
vector<int> res(n,0);
stack<int> st;
for(int i=0;i<n;i++) {
while(!st.empty()&&vec[st.top()]<vec[i]) {
res[st.top()]=i;
st.pop();
}
st.push(i);
}
for(auto r:res) {
cout<<r<<" ";
}
cout<<endl;
system("pause");
return 0;
}