在学校中,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)))