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

华为OD机试 2023B卷题库疯狂收录中，刷题点这里

专栏导读

本专栏收录于《华为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没有好朋友；

四、解题思路

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

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

五、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);
}

六、效果展示

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没有好朋友；

下一篇：华为OD机试 - 荒岛求生 - 栈Stack（Java 2023 B卷 100分）

本文收录于，华为OD机试（JAVA）真题（A卷+B卷）

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