【貪心】LeetCode 406. 根據身高重建隊列【中等】

假設有打亂順序的一羣人站成一個隊列,數組 people 表示隊列中一些人的屬性(不一定按順序)。每個 people[i] = [hi, ki] 表示第 i 個人的身高爲 hi ,前面 正好 有 ki 個身高大於或等於 hi 的人。

請你重新構造並返回輸入數組 people 所表示的隊列。返回的隊列應該格式化爲數組 queue ,其中 queue[j] = [hj, kj] 是隊列中第 j 個人的屬性(queue[0] 是排在隊列前面的人)。

 

示例 1:

輸入: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]] 是重新構造後的隊列。


示例 2:

輸入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
輸出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]
 

提示:

1 <= people.length <= 2000
0 <= hi <= 106
0 <= ki < people.length
題目數據確保隊列可以被重建


分析

題目簡單描述爲:整數對(h, k)表示,其中h這個人的身高,k是排在這個人前面的且身高大於或等於h的人數。

套路/思路:一般這種數對,還涉及排序的,根據第一個元素正向排序,根據第二個元素反向排序;或根據第一個元素反向排序,根據第二個元素正向排序,往往能夠簡化解題過程。

本題中,首先對數對進行排序,按照數對的第一個元素降序排序,按照數對的第二個元素升序排序。原因是:按照數對的第一個元素進行降序排序,對於每個元素,在其之前的元素的個數,就是大於等於它的元素的數量,而按照第二個元升序排序,我們希望k大的儘量排在後面,減少插入操作的次數

 

 

class Solution:
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        # 先排序,第一個身高元素降序排序,第二個元素升序排序。
        people = sorted(people, key = lambda x: (-x[0], x[1]) )
        res = []
        for p in people:
            if len(res) <= p[1]:
                res.append(p)
            elif len(res) > p[1]:
                res.insert(p[1], p) # 將p插入在res中index在p[1]的位置
        return res

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章