leetcode-215. 数组中的第K个最大元素

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个最大元素