算法通关村——解析堆在寻找第k大的应用

就拿力扣215题 数组中的第K个最大元素来说

1、我们可以用优先队列（PriorityQueue）建立一个找大小为k的最小堆

2、将数组中的前k个元素都放入最小堆

3、然后从第k个元素开始和堆顶比较，比堆顶小的丢掉；比堆顶大的，移除堆顶元素，将新来的元素放入堆中，一直遍历到剩下的元素都比较了结束

4、此时的堆顶就是我们要的第k大的元素

代码如下：

    public int findKthLargest(int[] nums, int k) {
        int len = nums.length;

        if(len < k){
            return -1;
        }

        PriorityQueue<Integer> minHeap = new PriorityQueue<>(k, (a,b) -> a - b);
        for(int i = 0; i < k; i++){
            minHeap.add(nums[i]);
        }
        for(int i = k; i < len; i++){
            Integer topEle = minHeap.peek();
            //堆顶和数组元素比较
            if(topEle < nums[i]){
                minHeap.poll();
                minHeap.offer(nums[i]);
            }
        }
        return minHeap.peek();
    }