问题描述
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示,其中h是这个人的身高,k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
注意:
总人数少于1100人。
示例:
输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
解题方法1
class Solution {
public int[][] reconstructQueue(int[][] people) {
// 先将数组排序,先按身高从高往低排序,然后K值从小到大排序,注意比较器的写法
Arrays.sort(people, new Comparator<int[]>() {
public int compare(int[] person1, int[] person2) {
if(person1[0] == person2[0]) {
return person1[1] - person2[1];
} else {
return person2[0] - person1[0];
}
}
});
// 将排好序的数组按k值进行插入
List<int[]> res = new ArrayList<>();
for (int[] person : people) {
res.add(person[1], person);
}
return res.toArray(new int[res.size()][2]);
}
}