在未排序的数组中找到第 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 ≤ 数组的长度。
答案:
1public int findKthLargest(int[] nums, int k) {
2 final int N = nums.length;
3 Arrays.sort(nums);
4 return nums[N - k];
5}
解析:
这种解法是最简单的,直接排序,然后再找第k个即可。但很明显这种解法在面试中不占优势,虽然做出来了,但感觉做和没做都一个样。下面再来看种解法
1public int findKthLargest(int[] nums, int k) {
2 final PriorityQueue<Integer> pq = new PriorityQueue<>();
3 for (int val : nums) {
4 pq.offer(val);
5 if (pq.size() > k) {
6 pq.poll();
7 }
8 }
9 return pq.peek();
10}
PriorityQueue是一种排序队列,offer表示往里面添加元素,添加的时候会进行排序,PriorityQueue默认是小顶堆的一种数据结构,最顶端的数据是最小的,poll表示的是移除最顶端的元素,也就是移除最小的元素。peek表示的是取出最顶端的元素,代码很好理解。这里我们使用的java的api,如果不让我们使用PriorityQueue类,我们该怎么办呢,其实也很简单,大家可以参照我之前写的106,排序-堆排序,自己建立一个大小为k的小顶堆即可。每次循环的时候如果堆的size小于k,直接add进去即可,如果等于k则要和堆最顶端的元素对比,如果比堆顶端的元素大,就把堆顶端的元素移除,然后把要比较的值放到堆的最顶端,让后再往下调整。代码就不再写了,有兴趣的可以自己去写写。