【leetcode-Python】-贪心-406. Queue Reconstruction by Height

题目链接

https://leetcode.com/problems/queue-reconstruction-by-height/

题目描述

假设一群人打乱顺序站成一个队列，数组 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个身高更高或相同的人在他前面。

在索引3位置的人身高为6，有1个身高更高或相同的人在他前面。

在索引4位置的人身高为4，有4个身高更高或相同的人在他前面。

在索引5位置的人身高为7，有1个身高更高或相同的人在他前面。

解题思路

对于这种给定数据为数对形式，还涉及到排序的问题，一般根据数对中第一个元素升序排序，在第一个元素相同的情况下根据第二个元素降序排序；或者根据第一个元素降序排序，在第一个元素相同的情况下根据第二个元素升序排序，这样往往能够简化解题过程。

由于每个人除了自己的身高之外，只记得前面身高比自己更高或相同的人数，因此我们可以先安排身高最高的人的位置，然后再安排身高低的人的位置。对于身高相等的情况，我们先安排k更小的人的位置。

那么在这个题中，我们首先按照第一个元素——身高对数组降序排序，然后按照第二个元素升序排序。对people就地排序过后，对于people[i]，索引小于i的人的身高都大于等于people[i]的身高。同时和people[i]身高相等且索引小于i的人，他们的k都小于people[i]的k。

我们维护一个返回列表result，并且从左到右遍历排序后的数组，将people[i]插入到对应的位置。当我们要放people[i]时，索引为0,...,i-1的人已经在列表中了（要么比people[i]高，要么身高相同，k比people[i].k小）。

索引为i+1,...,n-1的人还没有被放入result中，但他们要么比people[i]矮，插入位置不会对people[i]的插入造成任何影响，要么和people[i]一样高，但是一定会排在people[i]的后面（因为根据排序，他们的k比people[i]的key大），同样不会对people[i]的插入造成任何影响。

如果当前result中的元素个数小于等于people[i].k，那么people[i]先直接append到result中。如果当前result中的元素个数大于people[i].k，说明people[i]应该插入到result中适当的位置。由于当前result中的元素要么身高比people[i]高，要么身高相同且k比people[i].k小，并且people[i]前面要有k个身高更高或相同的人，因此需要将people[i]插入到result[k]的位置。

如示例中的people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]，就地排序后的结果为people = [[7,0],[7,1],[6,1],[5,0],[5,2],[4,4]]，初始化结果数组res。计算步骤如下所示：

Python实现

在Python中，实现按照第一个元素降序排序，第一个元素相同时第二个元素升序排序。可以设置sort函数中的key参数来实现。

list_object.sort(key=lambda x:(-x[0],x[1]))

另外选用insert()将对象插入到列表的指定位置（注意当列表为空时，位置参数无效），时间复杂度为O(N)：

#index为对象obj需要插入的索引位置，obj为待插入列表中的对象
list_object.insert(index,obj)

整体代码为：

class Solution:
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        res = []
        #就地排序，按照第一个元素降序排序，然后按照第二个元素升序排序
        people.sort(key = lambda x:(-x[0],x[1]))
        for p in people:
            if len(res)<=p[1]:
                res.append(p)
            else:#如果res中元素大于p的k(p[1])值，则将p插入到res[p[1]]的位置
                res.insert(p[1],p)
        return res
        
        
        

参考

https://leetcode-cn.com/problems/queue-reconstruction-by-height/solution/gen-ju-shen-gao-zhong-jian-dui-lie-by-leetcode-sol/

https://leetcode-cn.com/problems/queue-reconstruction-by-height/solution/xian-pai-xu-zai-cha-dui-dong-hua-yan-shi-suan-fa-g/

 扩展

【leetcode-Python】-贪心-135. Candy