2024华为OD机试真题-找朋友-(C++/Python)-C卷D卷-100分

 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;
}

python

n=int(input())
vec=list(map(int,input().split()))
res=[]
for i in range(n):
    res.append(0)
stack=[]
for i in range(n):
    while len(stack)>0 and