身高矮的可以插到身高高的前面去,因为矮的不会影响k值,高的会影响k值,对身高进行排序后,再根据k值进行插入,直接插入到第k个位置(对后面高的没影响)
class Solution {
public int[][] reconstructQueue(int[][] people) {
// sort()的第二个参数是comparator,可以通过lambda进行自定义比较
Arrays.sort(people, (a, b) -> {
// 对a[i][j],若与b[i][j]相等,j大的排后面
if (a[0] == b[0]) return a[1] - b[1];
// 若不相等,i大的(身高高的排前面)
return b[0] - a[0];
});
LinkedList<int[]> list = new LinkedList<>();
for (int i = 0; i < people.length; i++) {
// add()方法可以直接将元素插入指定位置 add(int index, int num),直接将num插入index位置
list.add(people[i][1], people[i]);
}
// toArray方法,直接将集合转变为数组
return list.toArray(new int[people.length][]);
}
}