0贪心算法中等 LeetCode406. 根据身高重建队列

406. 根据身高重建队列

描述

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

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

分析

• 按照身高由大到小排序，相同身高k值小的在前面。
• 将排好序的数组插入list集合，按照刚才的顺序插入，插入的规则是，插入到people[i][1]的位置，这样就满足了people[i][1]的含义：people[i][]前面有people[i][1]个比自己大于或等于的数组。
• 按照身高由大到小排序的好处是，再插入list集合时，比自己早先插入的肯定比自己要大,排序靠后的people插入的时候，即便按要求插入到了前面，也不影响已经插入的数组，因为后面插入的数组比已经插入的数组身高小，所以插入的时候满足people[i][1]的含义，结束的时候依旧满足。

插入利用了list集合add​(int index, E element)的性质，不用自己为了插入而移动数组元素。

public void add​(int index, E element)
将指定元素插入此列表中的指定位置。 将当前位置的元素（如果有）和任何后续元素向右移动（将其添加到索引中）。

为了不让list集合频繁扩容，应该在创建list的时候指定list集合大小。

class Solution {
    public int[][] reconstructQueue(int[][] people) {
		int n = people.length;
		//按照身高由大到小排序，相同身高k值小的在前面
        Arrays.sort(people,new Comparator<int[]>(){
           public int compare(int[] a, int[] b){
               if(a[0] == b[0]){
                   return a[1] - b[1];
               }else{
                   return b[0] - a[0];
               }
           }
        });
        ArrayList<int[]> list = new ArrayList<>(n);
        for(int[] arr : people){
            list.add(arr[1],arr);
        }
        int[][] res = new int[n][2];
        int i = 0;
        for(int[] li : list){
            res[i++] = li;
        }
        return res;
    }
}