题目描述

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

python

import sys

# 输入小朋友的数量
n = int(input())

# 输入小朋友的身高
heights = list(map(int, input().split()))

# 用于存储每个小朋友的好朋友位置
friendIndexes = [0] * n

# 使用栈来记录每个小朋友的位置
stack = [0]
for i in range(1, n):
    while stack and heights[i] > heights[stack[-1]]:
        # 如果当前小朋友的身高大于栈