题目3-python华为原题题库-找朋友

本篇题目：找朋友

题目描述

在学校中，N 个小朋友站成一队， 第 i 个小朋友的身高为 height[i]，

第 i 个小朋友可以看到的第一个比自己身高更高的小朋友 j，那么 j 是 i 的好朋友(要求 j > i)。

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

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

输入描述

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

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

输出描述

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

示例一

输入

2
100 95

输出

0 0

说明

第一个小朋友身高 100，站在队尾位置，向队首看，没有比他身高高的小朋友，所以输出第一个值为 0。

第二个小朋友站在队首，前面也没有比他身高高的小朋友，所以输出第二个值为 0。

示例二

输入

8
123 124 125 121 119 122 126 123

• 1
• 2

输出

1 2 6 5 5 6 0 0

• 1

说明

123 的好朋友是 1 位置上的 124

124 的好朋友是 2 位置上的 125

125 的好朋友是 6 位置上的 126

以此类推

代码实现

n = 8
sl = [123, 124, 125, 121, 119, 122, 126, 123]
hl = []
jl = [0] * n
for i, h in enumerate(sl):
    if not hl:
        hl.append([i, h])
        continue
    while hl and h > hl[-1][1]:
        ti, _ = hl.pop()
        jl[ti] = i
    hl.append([i, h])
print(*jl)