leetcode 406. 根据身高重建队列

最新推荐文章于 2024-07-05 20:18:25 发布

路漫途远

最新推荐文章于 2024-07-05 20:18:25 发布

阅读量112

点赞数

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/qq_41662990/article/details/108583005

版权

leetcode 专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一种解决特定队列重建问题的算法。问题设定为一群人的队列被打乱，每个人由一对数值(h,k)表示，h是身高，k是排在该人前面且身高大于或等于h的人数。文章详细解释了如何通过排序和插入操作，重建原始队列的算法思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

假设有打乱顺序的一群人站成一个队列。每个人由一个整数对(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]]

解法

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

这道题的难点在于想清楚该以什么样的顺序来进行排序。对于某个人来说，k的值只与比他还要高的人有关，比他低的人无论怎么排都不会影响到他的k，所以我们想到先对高的进行排序。现在假设每一个人的排序情况，已知一个待排的people，和一个已经排好序的队列，如果要对其进行插入，那么对于这个人而言队列里的每个人h都是>=他的h，所以我们可以根据他的k值直接在对应的位置进行插入，就能表示前面的几个比他高。最后需要注意到我们对第二位的排序方式应为升序。