Kth Largest Element in an Array

最新推荐文章于 2020-02-14 10:29:28 发布

原创最新推荐文章于 2020-02-14 10:29:28 发布 · 291 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

strStr

当前文章被收录于：

strStr

360 篇文章 0 人学习

订阅专栏查看详情

候着

当前文章被收录于：

候着

26 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种使用最小堆解决寻找数组中第K大的元素的问题。通过维护一个大小为K的最小堆，可以有效地找到目标元素。该方法优于直接排序，在大数据集上表现更佳。

这道题我们先用heap做了，之后再用排序的方法处理。

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> queue = new PriorityQueue<>();
    	for (int i : nums) {
    		queue.offer(i);
    		if (queue.size() > k) {
    			queue.poll();
    		} 
    	}
    	return queue.peek();
    }
}