牛客-NC88-寻找第K大

NC88. 寻找第K大(medium)


方法一:栈法

思路:同牛客-NC119-最小的K个数,求第K大等价于在小根堆中求第N-K+1小,所有改变遍历次数即可:

import java.util.*;

public class Solution {
   
    public int findKth(int[] a, int n, int K) {
   
        // write code here
        // 特判
        int ret = 0;
        if (n == 0 || K > n) {
   
            return ret;
        }

        PriorityQueue<Integer> pq = new PriorityQueue();
        for (int i = 0; i < n; i++) {
   
            pq.offer(a[i]);
        }
        for (int i = 0; i <= n - K; i++) {
   
            ret = pq.poll();
        }
        return ret;
    }
}

时间复杂度: O(NlogN),建小根堆所需的时间复杂度。
空间复杂度: O(N), 我们使用大小为N的PriorityQueue来存数据。

全部评论

相关推荐

酷酷我灵儿帅:这去不去和线不线下面说实话没啥关系
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 17:10
什么素质,我请问呢,要掉小珍珠了。。。又憋屈又生气
Steven267:这不喷回去?花钱是大爷,记住这个道理
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务