码迷,mamicode.com
首页 > 其他好文 > 详细

最大/最小de K个数/第K个数

时间:2020-02-24 00:25:13      阅读:93      评论:0      收藏:0      [点我收藏+]

标签:return   for   queue   最小   比较   数组   size   而不是   需要   

题目

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

思路

堆排序

收获

用优先队列实现最大最小堆
注意下列代码中优先队列的创建声明,需要自定义比较函数

代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> h = new PriorityQueue<>((n1,n2)->n1-n2);
        for(int i:nums){
            h.add(i);
            if(h.size()>k){
                h.poll();
            }
        }
        return h.poll();
    }
}

最大/最小de K个数/第K个数

标签:return   for   queue   最小   比较   数组   size   而不是   需要   

原文地址:https://www.cnblogs.com/Jun10ng/p/12355026.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!