找朋友--单调栈

在学校中，N个小朋友站成一队，第i个小朋友的身高为height[i]，第i个小朋友可以看到的第一比自己身高更高的朋友 i，那么 j 是i的好朋友(要求j > i)。
请输出一个数组，对应位置的输出是每个小朋友的好朋友位置，如果没有看到好朋友，则输出0。
输入描述:
第一行输入N，N表示有N个小朋友
第二行输入N个小朋友的身高height[i]，都是整数
输出描述:
输出N个小朋友的好朋友的位置
 

n = int(input())
height = list(map(int, input().split()))
stack = list()
ans = [0] * n
for i in range(n-1, -1, -1):
    h = height[i]
    while len(stack) > 0 and h >= height[stack[-1]]:
        stack.pop()
    if len(stack) > 0:
        ans[i] = stack[-1]
    stack.append(i)
print(" ".join(map(str, ans)))