题目描述
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
解题思路
为了使插入操作不影响后续的操作,身高较高的学生应该先做插入操作,否则身高较小的学生原先正确插入的第 k 个位置可能会变成第 k+1 个位置。
身高 h 降序、个数 k 值升序,然后将某个学生插入队列的第 k 个位置中。
高个子先站好位,矮个子插入到K位置上,因为已经插好的都是比它高的,所以数k个位置就行啦
而且矮个子 就算插到了已有高个子的前面也不会影响当前这个高个子的位置的正确性。
class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people,(new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if (o1[0] > o2[0]){
return -1;
}else if (o1[0] == o2[0]){
return o1[1] - o2[1];
}else{
return 1;
}
}
}));
LinkedList<int[]> list = new LinkedList<>();
for (int[] a : people) {
int index = a[1];
list.add(index, a);
}
return list.toArray(new int[list.size()][2]);
}
}