406. 根据身高重建队列

406. 根据身高重建队列

假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面 正好 有 ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。

答案

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

身高 h 降序、个数 k 值升序，然后将某个学生插入队列的第 k 个位置中。

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, (a, b) -> (a[0] == b[0] ? a[1] - b[1] : b[0] - a[0]));
        List<int[]> queue = new ArrayList<>();
        for (int[] p : people) {
            queue.add(p[1], p);
        }
        return queue.toArray(new int[queue.size()][]);
    }
}

自己思路
个数 k 值升序、身高 h 升序、将某个学生从前往后遍历，使得前面有k个不低于他的。

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, new Comparator<int[]>() {
            public int compare(int[] a, int[] b) {
                return a[1] == b[1] ? a[0] - b[0] : a[1] - b[1];
            }
        });
        int len = people.length;
        int[][] res = new int[len][2];
        res[0] = people[0];
        for (int i = 1; i < len; i++) {
            int[] p = people[i];
            int j = 0, k = 0;
            while (j < i) {
                if (res[j][0] >= p[0]) {
                    k++;
                }
                if (p[1] < k) {
                    break;
                }
                j++;
            }
            for (int q = i; q > j; q--) {
                res[q] = res[q - 1];
            }
            res[j] = p;
        }
        return res;
    }
}