JAVA解法
class Solution {
public int findKthLargest(int[] nums, int k) {
//定义优先队列(小根堆)
PriorityQueue<Integer> minTree = new PriorityQueue<>();
for (int i = 0; i < nums.length; i++) {
//每个数都加入到小根堆中。
minTree.add(nums[i]);
//如果加入之后堆的大小大于 k,就弹出堆顶。
if (minTree.size() > k) {
minTree.poll();
}
}
//堆顶的数就是第k大的。
return minTree.peek();
}
}
题解分析
这道题可以用优先队列建立小根堆,将所有数加入到小根堆中,并限制小根堆的个数为 k,因此最大的数在最下边,第 k 大的数就是堆顶的数,返回堆顶的数即可。
leetcode原题:215. 数组中的第K个最大元素
评论区