416.根据身高重建队列
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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值是说排在这个人前面和我一样高或者比我高的人数,也就是说身高矮的影响不到身高高的人,题解里有句话说得好,高个子看不到矮个子。
所以我们从身高高的人先开始排序,根据K值确定身高高的人的位置,再由身高由高到低将人插入进去。
class Solution406 {
public int[][] reconstructQueue(int[][] people) {
int[][] res = new int[people.length][people[0].length];
//按身高降序排列,身高相同按K值升序排列
Arrays.sort(people, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if(o1[0] == o2[0]) {
return o1[1] - o2[1];
}else {
return o2[0] - o1[0];
}
}
});
//将排好序的人按K值插入
LinkedList<int[]> list = new LinkedList<>();
for(int i = 0; i < people.length; i++) {
list.add(people[i][1], people[i]);
}
int i = 0;
for(Iterator iterator = list.iterator(); iterator.hasNext();) {
res[i] = (int[]) iterator.next();
i++;
}
return res;
}
}