美文网首页
力扣-215-数组中第K个最大元素

力扣-215-数组中第K个最大元素

作者: JoeyiDream | 来源:发表于2021-06-09 22:51 被阅读0次

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

    思路:最小堆维护k个最大元素。

    代码

    class Solution {
        public int findKthLargest(int[] nums, int k) {
            PriorityQueue<Integer> maxK = new PriorityQueue<Integer>((n1, n2) -> n1-n2);
            int len = nums.length;
            for (int i = 0; i < len; i++) {
                maxK.add(nums[i]);
                if (maxK.size() > k) {
                    maxK.poll();
                }
            }
            return maxK.poll();
        }
    }
    

    时间复杂度 O(nlogk) -------O(logk)是每次插入时堆排序的时间复杂度
    空间复杂度 O(k)

    相关文章

      网友评论

          本文标题:力扣-215-数组中第K个最大元素

          本文链接:https://www.haomeiwen.com/subject/wecseltx.html