Fix the content

This commit is contained in:
krahets
2023-08-10 11:35:16 +08:00
parent 932d14644d
commit 6ef4dc6444
10 changed files with 14 additions and 14 deletions

View File

@ -15,14 +15,14 @@ public class top_k {
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;