LeetCode215 数组的第k个最大元素

题目

给定整数数组 nums 和整数 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
提示：
1 <= k <= nums.length <= 104
-104 <= nums[i] <= 104

方法

暴力法

先排序再取第k大元素

• 时间复杂度：O（nlogn）
• 空间复杂度：O（）

class Solution {
    public int findKthLargest(int[] nums, int k) {
        Arrays.sort(nums);
        return nums[nums.length-k];
    }
}

最小堆法

创建容量为k的最小堆，若堆未满则添加，若堆满了，则比较堆顶最小值和数组当前值，若小于最小值则舍弃，如大于等于则去除堆顶加入当前值

• 时间复杂度：O（）
• 空间复杂度：O（）

class Solution {
    public int findKthLargest(int[] nums, int k) {
        Queue queue = new PriorityQueue<>();
        for(int num:nums){
            if(queue.size()queue.peek()){
                queue.offer(num);
                queue.poll();
            }
        }
        return queue.poll();
    }
}

分治法

根据快排的思想