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

题目

思路
1.用优先级队列实现容量为k的最小堆。
2.先把数组中前k个数offer进队列中
3.再遍历数组后面的数，若nums[i]比堆顶元素大，则移除堆顶元素，把nums[i]offer进数组。
4.遍历完数组后，堆顶元素就是第K大的数。

具体代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k,(o1,o2)->o1-o2);
        for(int i = 0;i < k;i++){
            priorityQueue.offer(nums[i]);
        }
        for(int i = k;i < nums.length;i++){
            if(nums[i]>priorityQueue.peek()){
                priorityQueue.poll();
                priorityQueue.offer(nums[i]);
            }
        }
        return priorityQueue.peek();
    }
}