【贪心法】Leetcode --- NO.406 根据身高重建队列（Java）

题目描述

题目分析

• 这道题就是将原本排好的队列在打乱后整理恢复原状
• 第一个数h代表身高，第二个数k代表这个人之前有几个身高更高或相等的人数

解法分析

• 可以知道，k只跟身高更高的人相关，跟身高较矮的人没有关系
• 所以可以先按照身高降序排序，如果身高相同就根据第二个数升序排序
• 那这样排好序之后，逐个遍历，放到结果队列中
• 因为身高已经是从高到低了，那么插入的时候主要是根据k
• 如果k=2，就代表前面有两个更高的人，假设之前已经插入了n个人了，这n个人肯定都是比当前这个人更高的，那么直接插入到索引为2的位置即可，即：0和1的身高比2更高或相等
• 可参考《官方题解》的算法图解

代码

class Solution {
    public int[][] reconstructQueue(int[][] people) {
    	// 判空
        if (people == null || people.length == 0 || people[0].length == 0) {
            return new int[0][0];
        }
        // 排序，建议不要使用lambda表达式
        Arrays.sort(people,(a,b) -> (a[0] == b[0] ? a[1]-b[1] : b[0]-a[0]) );
        // 使用List
        List<int[]> queue = new ArrayList<>();
        // 逐个遍历排好序的数组
        for(int[] p : people){
        	// 直接根据 k 插入即可，因为身高已经是排好序了的
            queue.add(p[1], p);
        }
        // 返回结果数组
        return queue.toArray(new int[queue.size()][]);
    }
}