LeetCode 406. 根据身高重建队列（Medium）

最新推荐文章于 2021-12-29 09:58:53 发布

原创最新推荐文章于 2021-12-29 09:58:53 发布 · 155 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

LeetCode 专栏收录该内容

100 篇文章

订阅专栏

这篇博客介绍了一种使用Python解决根据身高重建队列的算法问题。通过0114不一致排序和插入的方法，首先对输入的人员列表按身高降序和前面人数顺序进行排序，然后依次将排序后的人员插入到正确的位置，最终得到重建后的队列。代码简洁明了，包含两种插入方式实现。

在这里插入图片描述
【题目链接】

题解

根据身高重建队列

思路

在这里插入图片描述

代码

class Solution:
    ### 0114 不一致排序 + 插入（36 ms，15.4 MB）
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        # 先排序：第一位元素（身高）逆序，第二位元素（前面的人数）顺序
        people.sort(key=lambda x: (-x[0], x[1]))
        
        ans = []

        # 把排序后的每一个人根据 前面的人数 插入到对应位置
        for person in people:
            ## 写法1
            ans.insert(person[1], person)

            ## 写法2
            ans[person[1]:person[1]] = [person]

        return ans