Leetcode Link: 406. 根据身高重建队列 - 力扣(LeetCode)

题目

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

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

本题题目就不好理解

解法一

思路参考 题目描述:整数对 (h, k) 表示,其中 h 是这个人的身高,k 是排在这个人前面且身高大于或等于 h 的人数。

在本题目中,我首先对数对进行排序,按照数对的h降序排序,按照数对的k升序排序。原因是,按照h进行降序排序,对于每个元素,在其之前的元素的个数,就是大于等于他的元素的数量,而按照第二个元素正向排序,我们希望 k 大的尽量在后面,减少插入操作的次数。(参考链接中又ppt可以看,或者看下面的例子)

题解

class Solution:
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        new_people = sorted(people, key=lambda x: (-x[0], x[1]))
        res = []
        for val in new_people:
            res.insert(val[1], val)
 
        return res

启发和联系

一般这种数对,还涉及排序的,根据第一个元素正向排序,根据第二个元素反向排序,或者根据第一个元素反向排序,根据第二个元素正向排序,往往能够简化解题过程。

类似的,分发糖果中,是需要满足两个规则(两个维度),跟这个题目的思路都是一个一个维度(或者规则)满足