华为OD机试题-找朋友

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

请输出一个数组，对应位置的输出是每个小朋友的好朋友位置，如果没有看到好朋友，则输出0。

输入描述

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

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

输出描述

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

示例1
输入

2

100 95

输出

0 0

示例2
输入：

8
123 124 125 121 119 122 126 123

输出：

1 2 6 5 5 6 0 0

import java.util.*;

public class Main {
    public static void main(String[] args){
        /*  在学校中，N个小朋友站成一队，第i个小朋友的身高为height[i]，
        第i个小朋友可以看到的第一个比自己身高更高的小朋友j，
        那么j是i的好朋友(要求j>i)。
        请重新生成一个列表，对应位置的输出是每个小朋友的好朋友位置，
        如果没有看到好朋友，请在该位置用0代替。
         小朋友人数范围是 [0, 40000]。*/

        Scanner sc = new Scanner(System.in);
        int count = Integer.parseInt(sc.nextLine());//数组大小(小朋友个数)
        String[] split= sc.nextLine().split(" ");//每个数组元素

        Integer[] result = new Integer[count];//定义最后返回的集合
        //直接暴力解决
        for (int i = 0; i < count; i++) {
        //第一层遍历遍历每个小朋友身高
            int heightI = Integer.parseInt(split[i]);
            for (int j = i+1; j < count; j++) {
                //第二层遍历直接遍历第一层中小朋友位置之后的位置(寻找第一个比他高的)
                int heightJ = Integer.parseInt(split[j]);
                if (heightI < heightJ) {
                    result[i] = j;
                    break;
                }
            }
            //如果遍历到最后一个都没有比第一层遍历中大的就是直接返回0
            if (result[i] == null) {
                 result[i] = 0;
            }
        }
        for (Integer i : result) {
            System.out.print(i + " ");
        }
    }
}