406.根据身高重建队列

406.根据身高重建队列

题目

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

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

示例 1:

输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释:
编号为 0 的人身高为 5 ,没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ,有 2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

示例 2:

输入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]

提示:

1 <= people.length <= 2000
0 <= hi <= 106
0 <= ki < people.length
题目数据确保队列可以被重建

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queue-reconstruction-by-height
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

这里有两个维度h和k,**一定要想如何确定一个维度,然后按照另一个维度重新排序。

遇见两个维度权衡的时候,一定要先确定一个维度,再确定另外一个维度

这里选择先对身高从大到小排序(k表示的是前面比自己大的数个数)。

身高一样的按k从小到大排
举个例子,在身高一样,k不一样的时候,譬如[5,2]和[5,3], 对于最后排完的数组,[5,2]必然在[5,3]的前面。所以如果遍历的时候[5,3]在前面,等它先插入完,这个时候它前面会有3个大于等于它的数组对,遍历到[5,2]的时候,它必然又会插入[5,3]前面(因为它会插入链表索引为2的地方),这个时候[5,3]前面就会有4个大于等于它的数组对了,这样就会出错。

,这样比自己高就站在自己前面,就可以很清晰的得出站在前面人的个数。数组的索引与k就产生了一定的联系。

排好序之后一个维度就确定了,现在确定另外一个维度。从前往后进行最终位置确定,后面插入不会有影响前面的结果,因为后面的身高比前面的小,就算插在前面,他的身高也是比之前的小的,不会影响k值变化导致排序错误。

image

先确定一个维度,先对身高从大到小排序,身高一样的按k从小到大排。

  Arrays.sort(people, new Comparator<int[]>(){
            @Override
            public int compare(int[] o1,int[] o2){
                if(o1[0]==o2[0]){
                    return o1[1]-o2[1];
                }
              return o2[0]-o1[0];
            }
        });

排好序之后,开始调整顺序了,因为需要不断的插入数据,所以选择使用LinkedList。

  • ArrayList:List接口的主要实现类,线程不安全、效率高,底层使用Object[]存储,默认扩容为原数组的1.5倍
  • LinkedList:底层使用双向链表存储,对于频繁的插入、删除操作效率更高
  • Vector:List接口的古老实现类,线程安全、效率低,底层使用Object[]存储,默认长度为10。默认扩容为原数组的2倍
LinkedList<int[]> que = new LinkedList<>();
for(int[] p:people){
    que.add(p[1],p);
   }

结果返回的是二维数组,所以还需要讲LinkList转换成int[][]

集合转换成数组

coll.toArray();
toArray有两个重载的方法:
1.list.toArray():将list直接转为Object[] 数组;
2.list.toArray(T[] a):将list转化为你所需要类型的数组。

这里采用方法2,转换成我们需要的类型

return que.toArray(new int[people.length][]);

代码

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        if(people.length==1)return people;
        Arrays.sort(people, new Comparator<int[]>(){
            @Override
            public int compare(int[] o1,int[] o2){
                if(o1[0]==o2[0]){
                    return Integer.compare(o1[1],o2[1]);
                }
                return Integer.compare(o2[0],o1[0]);
            }
        }); 

         LinkedList<int[]> que = new LinkedList<>();

         for(int[] p:people){
             que.add(p[1],p);
         }

         return que.toArray(new int[people.length][]);
    }
}

套路
一般这种数对,还涉及排序的,根据第一个元素正向排序,根据第二个元素反向排序,或者根据第一个元素反向排序,根据第二个元素正向排序,往往能够简化解题过程。

posted @ 2021-07-30 21:38  rananie  阅读(41)  评论(0编辑  收藏  举报