LeetCode 406 根据身高和序号重组队列 java实现 算法之旅

根据身高和序号重组队列

Leetcode : 406. Queue Reconstruction by Height(Medium)

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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, k) 描述，h 表示身高，k 表示排在前面的有 k 个学生的身高比他高或者和他一样高。

解题思路：

为了在每次插入操作时不影响后续的操作，身高较高的学生应该先做插入操作，否则身高较小的学生原先正确插入第 k 个位置可能会变成第 k+1 个位置。

身高降序、k 值升序，然后按排好序的顺序插入队列的第 k 个位置中。

public int[][] reconstructQueue(int[][] people) {
    if(people == null || people.length == 0 || people[0].length == 0) return new int[0][0];

    Arrays.sort(people, new Comparator<int[]>() {
       public int compare(int[] a, int[] b) {
           if(a[0] == b[0]) return a[1] - b[1];
           return b[0] - a[0];
       }
    });

    int n = people.length;
    List<int[]> tmp = new ArrayList<>();
    for(int i = 0; i < n; i++) {
        tmp.add(people[i][1], new int[]{people[i][0], people[i][1]});
    }

    int[][] ret = new int[n][2];
    for(int i = 0; i < n; i++) {
        ret[i][0] = tmp.get(i)[0];
        ret[i][1] = tmp.get(i)[1];
    }
    return ret;
}

    public int[][] reconstructQueue(int[][] people) {
            if (people.length == 0 || people[0].length == 0)
                return new int[0][0];

            //根据身高降序排列，K值升值排列
            Arrays.sort(people, (a, b) ->  a[0] == b[0]? a[1] - b[1] : b[0] - a[0]);
            List<int[]> list = new ArrayList<>();
            //遍历people,添加进list
            for(int[] secondArray : people){
                list.add(secondArray[1], secondArray);
            }
            //插入完成， 输入到int[][]输出
            //将list变成二维数组
            return list.toArray(new int[list.size()][]);
    }

引用文献：https://blog.csdn.net/a724888/article/details/81389292#commentBox