题解 | 寻找第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) {
        // write code here
//小顶堆是从小到大排序的，要寻找第K大，所以将小顶堆的堆顶元素与下一位元素值进行比较，当下一位元素值小于堆顶元素的时候，直接丢掉，连堆都进不去，当下一位元素大于当前的堆顶元素的时候，舍弃堆顶元素，将下一个元素入堆，此时这个大小为 K 的小顶堆里，装的就是整个数组中最大的 K 个数字，所以堆顶即是我们需要的值
        PriorityQueue<Integer> min = new PriorityQueue<>();
        for(int i = 0;i<n;i++){
            if(min.size()< K)
            min.offer(a[i]);
            else{
                if(min.peek() < a[i]){
                    min.poll();
                    min.offer(a[i]);
                }
            }
        }
        return min.poll();
    }
}