【算法题】找朋友

共计 1096 个字符，预计需要花费 3 分钟才能阅读完成。

import java.util.Scanner;

/**
 * 标题：找朋友 | 时间限制：1秒 | 内存限制：262144K | 语言限制：不限
 * 在学校中，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
 * 输入
 * 8
 * 123 124 125 121 119 122 126 123
 * 输出
 * 1 2 6 5 5 6 0 0
 * 说明
 * 123的好朋友是1位置上的124
 * 124的好朋友是2位置上的125
 * 125的好朋友是6位置上的126
 * 以此类推
 *
 * @since 2022年5月1日
 */
public class M_N_T_20 {
    public static void main(String[] args) {

        Scanner scanner = new Scanner(System.in);
        int num = Integer.parseInt(scanner.nextLine());
        String[] heights = scanner.nextLine().split(" ");
        int[] friends = new int[num];
        StringBuilder stringBuilder = new StringBuilder();
        for (int i = 0; i < heights.length; i++) {
            int init = Integer.parseInt(heights[i]);
            friends[i] = 0;
            for (int j = i + 1; j < heights.length; j++) {
                int next = Integer.parseInt(heights[j]);
                if (next > init) {
                    friends[i] = j;
                    break;
                }
            }
            if (i == num - 1) {
                stringBuilder.append(friends[i]);
                break;
            }
            stringBuilder.append(friends[i]).append(" ");
        }

        System.out.println(stringBuilder);
    }
}