1. 程式人生 > 實用技巧 >根據身高重建佇列

根據身高重建佇列

根據身高重建佇列

題目:
假設有打亂順序的一群人站成一個佇列。 每個人由一個整數對(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]]

解題思路:先將陣列按照h進行降序排序,當h相等時按照k進行升序排序,之後依次按照元素的k進行插入即可

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, (a, b) -> {
            if(a[0] == b[0]) {
                return a[1] - b[1];
            }
            //降序排序
            return b[0] - a[0];
        });
        
        for(int i = 0; i < people.length; i++) {
            insert(people, i, people[i][1]);
        }
        
        return people;
    }
    
    private void insert(int[][] nums, int cur, int idx) {
        for(int i = cur; i > 0 && i > idx; i--) {
            int t[] = nums[i - 1];
            nums[i - 1] = nums[i];
            nums[i] = t;
        }
    }
}