题目描述:
思路:先排序后插入。
数组中每个元素的第一个元素为身高,即h,第二个元素为前面比他高或一样高的人,为k。
首先我们让这个数组先排好序,h降序,k升序。
排序后数组变为[[7,0],[7,1],[6,1],[5,0],[5,2],[4,4]](案例1)
接着我们一个一个插入新数组即可,每一个数插入到其k值得位置。
[[7, 0]]
[[7, 0], [7, 1]]
[[7, 0], [6, 1], [7, 1]]
[[5, 0], [7, 0], [6, 1], [7, 1]]
[[5, 0], [7, 0], [5, 2], [6, 1], [7, 1]]
[[5, 0], [7, 0], [5, 2], [6, 1], [4, 4], [7, 1]]
真的是非常神奇!为什么可以这样?
因为你会发现从高到矮得顺序插入的话,矮得插入对高的是没有影响的。因为对于k值来说,影响k值的只有比其高或者一样高的元素。(根据上面例子好好理解!)
代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people,(o1, o2) -> o1[0]==o2[0]?o1[1]-o2[1]:o2[0]-o1[0]);
//h从大到小,k从小到大
//插入k的位置即可
List<int []> list=new ArrayList<>();
for(int i=0;i<people.length;i++){
list.add(people[i][1],people[i]);
}
return list.toArray(new int[people.length][2]);
}
}