406. 根据身高重建队列
题目:
假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ,前面 正好 有 ki 个身高大于或等于 hi 的人。
请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。
思路:
本题有两个维度,h和k,看到这种题目一定要想如何确定一个维度,然后在按照另一个维度重新排列。如果两个维度一起考虑一定会顾此失彼。
有一个困惑的点就是先按h排序呢,还是先按照k排序呢?
如果按照k来从小到大排序,排完之后,会发现k的排列并不符合条件,身高也符合条件,两个维度哪一个都没确定下来。
那么按照身高h来排序呢,身高一定从大到小排(身高相同的话则k小的站前面),让高个子在前面。
此时我们可以确定一个维度了,就是身高,前面的节点一定都比本节点高!
那么只需要按照k为下标重新插入队列就可以了,如下图:
代码:
class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
// 先按身高降序,若身高相同则按 k 值升序。
if (o2[0]==o1[0])
return o1[1]-o2[1];
return o2[0]-o1[0];
}
//或
//return o1[0] == o2[0] ? o1[1] - o2[1] : o2[0] - o1[0];
});
//或Arrays.sort(people, (o1, o2) -> o1[0] == o2[0] ? o1[1] - o2[1] : o2[0] - o1[0]);
LinkedList<int[]> list = new LinkedList<>();
//遍历排序后的数组,根据 k 插入到 k 的位置上。
for(int [] k : people){
list.add(k[1] , k);
}
//调用list直接转为Object[] 数组;
return list.toArray(new int[list.size()][2]);
}
}
转载:
如有冒犯请私信我删除
链接:https://leetcode.cn/problems/queue-reconstruction-by-height/solution/406du-shuo-shi-tan-xin-na-yao-wei-shi-yao-yong-tan/