2022.4.18 力扣406

学习：贪心算法

follow ：代码随想录

---

406 根据身高重建队列

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

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

解析：
贪心思想：
将数组按照身高进行排序，之后按照身高的顺序遍历数组，确定每一个元素是否满足，直到找到第一个满足条件的元素

public int[][] reconstructQueue(int[][] people) {
        //将people按身高排序
        Arrays.sort(people, (o1, o2) -> (o1[0] - o2[0]));
        boolean[] flag = new boolean[people.length];
        Arrays.fill(flag, false);
        int[][] queue = new int[people.length][people[0].length];
        for(int i = 0; i < queue.length; i++){
            for(int j = 0; j < people.length; j++){
                //满足的条件
                if(!flag[j]){
                    int count = 0;
                    for(int k = 0; k < i; k++){
                        if(queue[k][0] >= people[j][0]){
                            count++;
                        }
                    }
                    if(count == people[j][1]){
                        queue[i] = people[j];
                        flag[j] =  true;
                        break; 
                    }
                }
            }
        }
        return queue;
    }

贪心思想：
将people按身高由大到小排序,身高相同的按照k从小到大排序，之后按照顺序元素的每一个k值插入到数组中，可以保证每个元素在插入时，前面的元素都是比他大的，所以只需要考虑k值即可

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        //将people按身高由大到小排序,身高相同的按照k从小到大排序
        Arrays.sort(people, (o1, o2) ->{
            if(o1[0] == o2[0]) return o1[1] - o2[1];
            else return o2[0] - o1[0];
        });
        //使用链表，效率更高
        List<int[]> que = new LinkedList<>();
        //之后按照身高由大到小的顺序，按照k的值插入
        //比如[5,2]即越过两个比5大的数，再插入
        for(int i = 0; i < people.length; i++){
            que.add(people[i][1], people[i]);
        }
        //链表转数组 toArray 指定类型
        return que.toArray(new int[people.length][]);
    }
}