刷题记录：华为OD机试题：找到好朋友

参考代码：(4条消息) 华为OD机试题：找到好朋友_CoCo_2022的博客-CSDN博客

题目：

题目描述：

在学校中，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，依此类推

---

自己的代码

while True:
    try:
        n=int(input())
        height = list(map(int,input().split()))
        friend = []
        for i in range(n):
            for j in range(i+1,n):
                if height[j]>height[i]:
                    friend.append(j)
                    break
            else:
                friend.append(0)
        print(' '.join(map(str,friend)))
    except Exception as e:
        break