LeetCode日记——【算法】排序专题

  题1：数组中的第k个最大元素（Kth Largest Element in an Array）

Leetcode题号：215

难度：Medium

链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array/

题目描述：

在未排序的数组中找到第 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 ≤ 数组的长度。

代码：

方法一：排序

1 class Solution {
2     public int findKthLargest(int[] nums, int k) {
3         Arrays.sort(nums);
4         return nums[nums.length-k];
5     }
6 }

方法二：堆排序

 1 class Solution {
 2     public int findKthLargest(int[] nums, int k) {
 3         //创建一个小顶堆
 4         PriorityQueue <Integer> pq = new PriorityQueue<>();
 5         //对nums数组中的每个元素进行遍历
 6         for(int val:nums){
 7             //把新元素加入到小顶堆中
 8             pq.add(val);
 9             //保持小顶堆大小为k
10             if(pq.size()>k){
11                 //排序后，删除堆顶元素（堆中最小的数），保证剩下的是遍历到的数中最大的k个数
12                 pq.poll();
13             }
14         }
15         //返回堆顶元素（即最大的k个数中最小的那个，也就是第k大的数）
16         return pq.peek();
17     }
18 }