题解 | #寻找第K大#

寻找第K大

https://www.nowcoder.com/practice/e016ad9b7f0b45048c58a9f27ba618bf

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) {
        PriorityQueue<Integer> queue = new PriorityQueue(K);
        for (int x : a) {
            if(queue.size()<K){
                queue.add(x);
            }else{
                if(x>queue.peek()){
                    queue.poll();
                    queue.add(x);
                }
            }
        }
        return queue.isEmpty() ? 0 : queue.peek();
    }
}

全部评论

相关推荐

头像
05-12 09:14
点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务