欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python语言程序设计(嵩天老师)-期末考试3.2-站队顺序输出

程序员文章站 2022-05-19 14:23:18
...

这道题的原题是Leetcode 406题

题目:假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h,k)表示,其中h是这个人的身高,k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。

思路:输入是二维列表,(1)根据列表的第二维进行排序,(2)在根据列表的第一维倒序排列,(3)最后根据列表第二维的值将排序后的序列插入新的列表中。

代码:

from operator import itemgetter
queue = eval(input())

queue.sort(key = itemgetter(1))
#print(queue)
queue.sort(key = itemgetter(0), reverse = True)
#print(queue)

output = []
for item in queue:
    output.insert(item[1], item)
    #print(output)
print(output)
参考博客:https://blog.csdn.net/sinat_27421407/article/details/80822444
相关标签: Python