Leetcode--Java--406. 根据身高重建队列

题目描述

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

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

样例描述

思路

贪心 + 套路排序 （排序 + 插队）
对数对的题，可以考虑按照第一个元素正向排序，第二个逆向排序，或者反之。

1. 本题按照第一个元素降序排序，第二个元素升序排序，因为对于大的数，比它大的数应该越少
2. 排好队之后要进行插入操作，比如某些元素前面比它大的很少，要往前移动
3. list将某数插入到指定位置的方法add(index, num)
4. list转化为数组的方法，list.toArray(new int[list.size()])
5. 比较时，重写new Comparator接口的compare函数，或者直接用键函数

代码

class Solution {
      public int[][] reconstructQueue(int[][] people) {
        //按数组第一个元素进行降序，按第二个元素进行升序
        //重写compare形式
        // Arrays.sort(people, new Comparator() {
        //     @Override
        //     public int compare(int[] person1, int[] person2){
        //         if (person1[0] != person2[0]){
        //            
        //             return person2[0] - person1[0];
        //         }else{
        //            
        //             return person1[1] - person2[1];
        //         }
        //     }
        // });
        //键函数简单明了
        Arrays.sort(people, (a, b) -> {
            if (a[0] != b[0]) {
                return b[0] - a[0];
            } else return a[1] - b[1];
        });
        //新建一个list,用于保存结果集
        List<int[]> list = new LinkedList<>();
        for (int i = 0; i < people.length; i++) {
            if (list.size() > people[i][1]){
                //结果集元素个数大于第i个人前面应有的人数时，将第i个人插入到结果集的 people[i][1]位置
                list.add(people[i][1],people[i]);
            }else{
                //结果集元素个数小于等于第i个人前面应有的人数时，将第i个人追加到结果集的后面
                list.add(list.size(),people[i]);
            }
        }
        //将list转化为数组，然后返回
        return list.toArray(new int[list.size()][]);
    }
}