leetcode406.根据身高重建队列 Python

最新推荐文章于 2025-05-20 02:30:00 发布

小栗子pola

最新推荐文章于 2025-05-20 02:30:00 发布

阅读量533

点赞数

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

本文链接：https://blog.youkuaiyun.com/qq_36849855/article/details/119837592

版权

leetcode 专栏收录该内容

129 篇文章

订阅专栏

该博客探讨了一种使用贪心算法解决队列重构的问题。具体来说，给定一组身高和前置高个子人数的数据，我们需要重构队列，使得每个人前面恰有k个身高不低于自己的人。首先对数据进行排序，然后从前往后依次插入队列，确保局部最优以达到全局最优。博客详细解释了算法思路和实现方法，并提供了具体的示例和代码实现。

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

题目：

假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面正好有 ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。

示例：

输入：people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出：[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释：

编号为 0 的人身高为 5 ，没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ，没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ，有 2 个身高更高或者相同的人排在他前面，即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ，有 1个身高更高或者相同的人排在他前面，即编号为 1 的人。
编号为 4 的人身高为 4 ，有 4 个身高更高或者相同的人排在他前面，即编号为0、1、2、3 的人。
编号为 5 的人身高为 7 ，有 1 个身高更高或者相同的人排在他前面，即编号为 1 的人。
因此[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

思路：

先排序再插入，插入的过程是贪心的过程，贪心就是保持每一阶段的局部最优，最后达到全局最优。

先按照身高由高到低排列，所以身高的索引加了负号，身高相同的按k排列，大的放后边，k表示的是前面有k个身高大于或者等于h的人。按身高排好以后，从前到后遍历，将数组插入到合适的位置，先实现局部最优，再达到全局最优，最终构造完成队列。

insert函数，参数1表示插入的位置，参数2表示插入的元素。
lambda，匿名函数，正常函数的简化写法，不需要对函数命名。

时间复杂度是n的平方，因为排序的时间复杂度是nlogn，插入的过程需要遍历全部数组，并且插入一个数需要移动其他数，所以时间复杂度是n的平方，它大于排序的时间复杂度。
空间复杂度来自排序过程使用的栈空间。

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]))
        queue = []
        for p in people:
            queue.insert(p[1],p)
        return queue