数组中的第K个最大元素

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

这道题最简单的方法肯定就是用Arrays.sort()来做，不过这样不仅时间空间复杂度高，而且面试也拿不出手。
这里利用堆，可以巧妙解决。
由于对堆不是很熟悉，这里是第一次使用。思路就是：建一个小顶堆，将所有数组中的元素加入堆中，并保持堆的大小小于等于 k。这样，堆中就保留了前 k 个最大的元素。这样，堆顶的元素就是正确答案。

• 时间复杂度：向大小为 k 的堆中添加元素的时间复杂度为 O(logk)，我们将重复该操作 N 次，故总时间复杂度为O(Nlogk)。
• 空间复杂度：O(k)。

代码如下：

class Solution {
    public int findKthLargest(int[] nums, int k) {
        // 初始化堆，使用优先级队列，小的元素先出队
        PriorityQueue<Integer> heap =
            new PriorityQueue<Integer>((n1, n2) -> n1 - n2);

        // 保证前K个大的元素在堆中
        for (int n: nums) {
          heap.add(n);
          if (heap.size() > k)
            heap.poll();
        }

        // 输出堆顶
        return heap.poll();        
  }
}