华为OD机试 - 找朋友（Java 2023 B卷 100分）

在这里插入图片描述

专栏导读

本专栏收录于《华为OD机试（JAVA）真题（A卷+B卷）》。

刷的越多，抽中的概率越大，每一题都有详细的答题思路、详细的代码注释、样例测试，发现新题目，随时更新，全天CSDN在线答疑。

一、题目描述

在学校中，N个小朋友站成一队，第i个小朋友的身高为height[i]，第i个小朋友可以看到的第一个比自己身高更高的小朋友j，那么是的好朋友(要求j>i)。

请重新生成一个列表，对应位置的输出是每个小朋友的好朋友位置，如果没有看到好朋友，请在该位置用0代替。

小朋友人数范围是[0,40000]

二、输入描述

第一行输入N，N表示有N个小朋友

第二行输入N个小朋友的身高height[i]，都是整数

三、输出描述

输出N个小朋友的好朋友的位置。

N个小朋友站成一队，第i个小朋友的身高为height[i]，第i个小朋友可以看到的第一个比自己身高更高的小朋友j，那么是的好朋友(要求j>i)。

大白话解释一下就是：

有一行数，找到比当前数大的第一个数，输出其下角标，如果没有，输出0

用例：

1、输入：

8
123 124 125 121 119 122 126 123

2、输出：

1 2 6 5 5 6 0 0

3、说明

123的好朋友是1位置上的124；
124的好朋友是2位置上的125；
125的好朋友是6位置上的126；
121的好朋友是5位置上的122；
119的好朋友是5位置上的122；
122的好朋友是6位置上的126；
126没有好朋友；
1223没有好朋友；

四、解题思路

本题还是比较简单的，理解题意才是关键。

输入 N个小朋友；
输入N个小朋友的身高；
非法校验；
遍历N个小朋友的身高；
- 找到比当前身高大的第一个身高，并获取其下角标；
- 如果未找到，则默认0；
按照指定格式输出。

五、Java算法源码

public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    // N个小朋友
    int n = Integer.valueOf(sc.nextLine());
    // N个小朋友的身高
    int[] arr = Arrays.stream(sc.nextLine().split(" ")).mapToInt(Integer::parseInt).toArray();

    // 非法校验
    if (n < 0 || n > 40000 || n != arr.length) {
        System.out.println("input error");
        return;
    }

    // 遍历N个小朋友的身高
    List<Integer> list = new ArrayList<>();
    for (int i = 0; i < n; i++) {
        boolean flag = false;
        // 找到比当前身高大的第一个身高，并获取其下角标
        for (int j = i; j < n; j++) {
            if (arr[j] > arr[i]) {
                list.add(j);
                flag = true;
                break;
            }
        }
        // 如果未找到，则默认0
        if (!flag) {
            list.add(0);
        }
    }

    StringJoiner stringJoiner = new StringJoiner(" ");
    list.forEach(x->stringJoiner.add(String.valueOf(x)));
    System.out.println(stringJoiner);
}