【LeetCode】406. 根据身高重建队列

问题描述

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/