问题描述
Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k)
, where h
is the height of the person and k
is the number of people in front of this person who have a height greater than or equal to h
. Write an algorithm to reconstruct the queue.
Note:
The number of people is less than 1,100.
假设你有一列随机的人在排队。每个人由一对整数(h, k)描述,其中h是这个人的高度,k是这个人前面的人的高度大于或等于h的人数。编写一个算法来重构队列。
注意:
人数不到1100人。
输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
Python 实现
实现一:使用堆。
根据 h 和 k 的大小构建堆,按照 h 的倒序排序,h 相同时,按照 k 的升序排序。需要说明的一点是,完成排序后,在往返回的列表使用 insert 时,pos 的值保证了当前插入的数对 h 必然比该位置后面已经插入的数对的 h 小(因为 h 的较大的数对已经插入了)。对于与其 h 值相等的数对,当前 pos 值必然较大,因此肯定会插入到相同 h 值的数对的后面。
class Solution(object):
def reconstructQueue(self, people):
"""
:type people: List[List[int]]
:rtype: List[List[int]]
"""
queue_heap = []
for p in people:
heapq.heappush(queue_heap, (-p[0], p[1]))
res = []
while len(queue_heap) > 0:
height, pos = heapq.heappop(queue_heap)
if pos >= len(res):
res.append([-height, pos])
else:
res.insert(pos, [-height, pos])
return res
实现二:使用 python 内置的排序,实现原理与上面类似。
class Solution(object):
def reconstructQueue(self, people):
"""
:type people: List[List[int]]
:rtype: List[List[int]]
"""
queue = sorted(people, key = lambda x: (-x[0], x[1]))
res = []
for p in queue:
res.insert(p[1], p)
return res
链接:https://leetcode.com/problems/queue-reconstruction-by-height/