406. 根据身高重建队列

最新推荐文章于 2024-11-14 23:35:31 发布

chandelierds

最新推荐文章于 2024-11-14 23:35:31 发布

阅读量74

点赞数

分类专栏： leetcode 文章标签： leetcode 数据结构算法

本文链接：https://blog.csdn.net/chandelierds/article/details/128754675

版权

leetcode 专栏收录该内容

123 篇文章 0 订阅

订阅专栏

该算法首先按身高h降序排列，同身高时，按k值（前面比自己高的人数）升序排列。然后遍历排序后的列表，将每个人插入到队列中对应的位置，确保前面有k个比自己高的。这样能保证高个子的相对顺序不变，不影响最终结果。

摘要由CSDN通过智能技术生成

在这里插入图片描述
思路：先按照身高h倒排，h相同，k大的往后排，然后从高到矮遍历，根据k插到相应的位置
为什么这么做：先把高个的排好了，相对位置也就基本确定了，矮个的无论怎么排，都不影响高个的位置

class Solution:
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        # 根据h倒排，h相同时，k大的往后排
        people.sort(key = lambda x: (-x[0], x[1]))
        # print(people)
        queue = []
        # 根据k插入
        for p in people:
            queue.insert(p[1], p)  # 插到k位置上，表示前面有k个比我高的
        return queue