leetcode 406
题目
假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi,
ki] 表示第 i 个人的身高为 hi ,前面 正好 有 ki 个身高大于或等于 hi 的人。请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj,
kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。示例 1:
输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 解释: 编号为 0 的人身高为 5
,没有身高更高或者相同的人排在他前面。 编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。 编号为 2 的人身高为 5 ,有
2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。 编号为 3 的人身高为 6 ,有 1
个身高更高或者相同的人排在他前面,即编号为 1 的人。 编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为
0、1、2、3 的人。 编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。 因此
[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queue-reconstruction-by-height
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解析
先按照身高降序排序,再按照k升序排序。在搞个循环插入。
原因:首先最后结果总体来说应该是高的在前面,矮的在后面。
如果高度相同,要让小的在前面,以免插入混乱。例如
7,0 6,1 5,0 5,2 如果排序成为7,0 6,1 5,2,5,0,
在考虑50时,为 70 61 52 将50插入到开头,52就会有问题。这就是要先考虑k小的
代码
class Solution {
public int[][] reconstructQueue(int[][] people) {
int lenth=people.length;
Arrays.sort(people,new Comparator<int[]>(){
public int compare(int[]a,int []b){
if(a[0]==b[0]){
return a[1]-b[1] ;
}else{
return b[0]-a[0];
}
}
});
List<int[]>res=new ArrayList<int[]>();
res.add(people[0]);
for(int i=1;i<lenth;i++){
if(res.size()>people[i][1]){
res.add(people[i][1],people[i]);
}else{
res.add(people[i]);
}
}
return res.toArray(new int[lenth][2]);
}
}