根据身高重建队列
难度:
标签:
题目描述
假设有打乱顺序的一群人站成一个队列,数组 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]] 是重新构造后的队列。
示例 2:
输入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]] 输出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]
提示:
1 <= people.length <= 2000
0 <= hi <= 106
0 <= ki < people.length
- 题目数据确保队列可以被重建
代码结果
运行时间: 24 ms, 内存: 16.5 MB
/*
* Problem Description:
* Given a list of people with their heights and the number of people in front of them who have a height greater than or equal to theirs, reconstruct the queue.
* Approach:
* 1. Use Java Stream to sort and insert elements.
* 2. The sorting and insertion follow the same logic as in the traditional method.
*/
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
public class Solution {
public int[][] reconstructQueue(int[][] people) {
// Sort the people array using Java Stream
List<int[]> sortedPeople = Arrays.stream(people)
.sorted((a, b) -> {
if (a[0] == b[0]) {
return a[1] - b[1];
} else {
return b[0] - a[0];
}
})
.collect(Collectors.toList());
// Create a list to hold the final queue
List<int[]> queue = new ArrayList<>();
// Insert each person into the queue at the index specified by their k-value
sortedPeople.forEach(person -> queue.add(person[1], person));
// Convert the list back to an array
return queue.toArray(new int[queue.size()][]);
}
}
解释
方法:
这个题解的思路是先对输入的人员数组 people 进行排序,排序规则是按照身高 h 降序,如果身高相等则按照 k 值升序。排序后,从前往后遍历排序后的数组,根据每个人的 k 值,将其插入到结果数组 que 中的第 k 个位置。最后得到的 que 数组即为重建后的队列顺序。
时间复杂度:
O(n^2)
空间复杂度:
O(n)
代码细节讲解
🦆
为什么在排序时选择身高降序、k值升序作为排序规则,而不是其他顺序?
▷🦆
如果`people`数组非常大,这种插入方法效率如何,有没有更优的方法来处理大数据量的情况?
▷🦆
这种算法对于所有可能的输入都是稳定的吗?还是说在某些特定的情况下,它可能会导致不正确的结果?
▷相关问题
计算右侧小于当前元素的个数
给你一个整数数组 nums
,按要求返回一个新数组 counts
。数组 counts
有该性质: counts[i]
的值是 nums[i]
右侧小于 nums[i]
的元素的数量。
示例 1:
输入:nums = [5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
示例 2:
输入:nums = [-1] 输出:[0]
示例 3:
输入:nums = [-1,-1] 输出:[0,0]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104