*LeetCode-Kth Largest Element in an Array

最新推荐文章于 2021-06-15 11:02:29 发布

原创最新推荐文章于 2021-06-15 11:02:29 发布 · 387 阅读

0 ·

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

leetcode 同时被 3 个专栏收录

220 篇文章

订阅专栏

queue

6 篇文章

订阅专栏

heap

2 篇文章

订阅专栏

本文介绍了一种使用最大堆寻找数组中第K大的元素的方法。通过维护一个大小为n-k+1的最大堆，确保堆中始终保存了数组中的较小部分元素。文章详细解释了如何在遍历数组过程中调整堆，并给出了完整的Java代码实现。

三种方法 1 排序 2 min heap 3 max heap

我用了3 其实2更直观只是要想清楚

因为heap是用priority queue实现的每次只能得到queue head 所以要是min 就要使用k个记录的是最大的k个

要是max就要使用 n - k + 1个记录的是最小的几个

再有就是comparator的实现注意返回值的意义只要是升序的那么就返回 int1 - int2

降序就是 int2 - int1

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        int n = nums.length;
        PriorityQueue<Integer> maxHeap = new PriorityQueue ( n - k + 1, new Comparator<Integer>(){
                public int compare ( Integer i, Integer j ){
                    return j - i;
                }
            });
        for ( int i = 0; i < n - k + 1; i ++ )
            maxHeap.offer(nums[i]);
        for ( int i = n - k + 1; i < n; i ++ ){
            if ( nums[i] < maxHeap.peek() ){
                maxHeap.poll();
                maxHeap.offer( nums[i] );
            }
        }
        return maxHeap.peek();
    }
}