华为od机试76 python找到好朋友

在学校中
  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
        #这里感觉写复杂了
        ans=''.join(str(x) for x in friend)
        #这里最后只得到‘126556‘所以为了补齐，选择用ljust
        print(' '.join(ans.ljust(N,'0')))
    except:
        break

中间感觉写的有点复杂了，结果还是好的
测试案例通过：