LeetCode 每日一题406. 根据身高重建队列

406. 根据身高重建队列

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]

解题思路：

对于两个维度的排序，通常需要先搞定一个维度的排序，再搞定另外一个。在不明确先排序哪个维度时，可以用穷举法。试出 身高排倒序，若身高相同则 k 小的排前面。

初始值：[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
排序后：[7,0], [7,1], [6,1], [5,0], [5,2]，[4,4]]

这样排序后：后面的人总比前面的人矮 或 k值大于前面的人。

从左往右遍历，将元素插入 k 的位置即完成排序。

因为后面的人总比前面的人矮，所以后面的人插入到前面时，并不会影响之前已经排序好的结果。

插入过程：

排序完成：[7,0], [7,1], [6,1], [5,0], [5,2]，[4,4]]
插入[7,0]：[[7,0]]
插入[7,1]：[[7,0],[7,1]]
插入[6,1]：[[7,0],[6,1],[7,1]]
插入[5,0]：[[5,0],[7,0],[6,1],[7,1]]
插入[5,2]：[[5,0],[7,0],[5,2],[6,1],[7,1]]
插入[4,4]：[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]

方法一：身高排倒序 + 插入法

public int[][] reconstructQueue(int[][] people) {
    // 身高相同，k小的排前面；身高不同，高的排前面
    Arrays.sort(people, (p1, p2) -> {
        if (p1[0] == p2[0]) {
            return p1[1] - p2[1];
        } else {
            return p2[0] - p1[0];
        }
    });

    // 根据 k 插入对应位置，LinkedList链表结构插入会快一些
    List<int[]> list = new LinkedList<>();
    for (int i = 0; i < people.length; i++) {
        // k 即为插入位置，因为后面的人总比前面的人矮，往前插入并不会影响之前已经排好的结果
        int k = people[i][1];
        list.add(k, people[i]);
    }

    return list.toArray(people);
}

执行结果：