leetcode 215. 数组中的第K个最大元素

2023.11.7

leetcode 215. 数组中的第K个最大元素_第1张图片

         本题两个思路,一个是先将数组排序,再从后往前数第k个元素即为第k个最大的元素。 或者使用优先队列,将数组元素放入队列中,当队列元素数量超过k时,弹出队列元素,由于优先队列的性质,每次弹出的都是最小的那个元素。 遍历完最后队列的第一个元素就是第k个最大的元素。

        java代码如下:

快排:

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

优先队列:

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

你可能感兴趣的:(leetcode专栏,leetcode,算法,java,数据结构)