假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
思路:首先我们给队列先排个序,按照身高高的排前面,如果身高相同,则第二个数小的排前面。然后我们新建一个空的数组,遍历之前排好序的数组,然后根据每个元素的第二个数字,将其插入到res数组中对应的位置
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people, new Comparator<int[]>(){
public int compare(int[] a, int[] b){//按身高降序排序(h大的在前面),按k的大小升序排列(k小的在前面)
if(a[0] != b[0]){
return b[0] - a[0];
}else{
return a[1] - b[1];
}
}
});
List<int[]> res = new LinkedList<>();
for(int i = 0; i < people.length; i++){
int[] peo = people[i];
res.add(peo[1], peo);
}
return res.toArray(new int[people.length][]);
}