题解 | #寻找第K大#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param a int整型一维数组
     * @param n int整型
     * @param K int整型
     * @return int整型
     */
    public int findKth(int[] a, int n, int K) {
        if (n == 0) {
            return 0;
        }
        // 创建一个容量为k的小顶堆队列，并初始化
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        for (int i = 0; i < K; i++) {
            queue.offer(a[i]);
        }

        // 将数组剩余元素遍历，该元素大于堆顶元素时，弹出堆顶元素，放入该数组元素
        for (int i = K; i < n; i++) {
            if (queue.peek() < a[i]) {
                queue.poll();
                queue.offer(a[i]);
            }
        }

        // 遍历完成后，堆顶的元素就是第K大
        return queue.peek();
    }
}