LeetCode-406. 根据身高重建队列 Python3版本

406. 根据身高重建队列


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

注意:
总人数少于1100人。

示例

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

解题思路:建立一个新列表,将people按照h从大到小,k从小到大的顺序进行插入。这样做的理由是先插入身高h大的,后插入身高h小的可以使得后面的插入对前面已经排序的不产生影响;同等身高前提下,先插入人数k小的,后插入人数k大的可以使得后面的插入对前面已经排序的不产生影响。

这里使用了一个python中的高级语法:key=lambda p:(-p[0],p[1]),其含义是如果第一个排序条件相同,则按照第二个条件排序。

Python3代码如下:

class Solution(object):
    def reconstructQueue(self, people):
        """
        :type people: List[List[int]]
        :rtype: List[List[int]]
        """
        result = []
        people.sort(key=lambda p:(-p[0],p[1]))
        for p in people:
            result.insert(p[1],p)
        return result
        

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值