题目:
假设有打乱顺序的一群人站成一个队列,数组 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