This commit is contained in:
krahets
2023-08-10 11:35:41 +08:00
parent 6d41b5da04
commit 059a849167
5 changed files with 14 additions and 14 deletions
+2 -2
View File
@@ -82,14 +82,14 @@ comments: true
Queue<Integer> heap = new PriorityQueue<Integer>();
// 将数组的前 k 个元素入堆
for (int i = 0; i < k; i++) {
heap.add(nums[i]);
heap.offer(nums[i]);
}
// 从第 k+1 个元素开始,保持堆的长度为 k
for (int i = k; i < nums.length; i++) {
// 若当前元素大于堆顶元素,则将堆顶元素出堆、当前元素入堆
if (nums[i] > heap.peek()) {
heap.poll();
heap.add(nums[i]);
heap.offer(nums[i]);
}
}
return heap;