首页 > 其他 > 详细

215. Kth Largest Element in an Array

时间:2016-09-25 07:23:18      阅读:219      评论:0      收藏:0      [点我收藏+]

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note: 
You may assume k is always valid, 1 ≤ k ≤ array‘s length.

思路:用priorityqueue建maxheap然后读出第几个。

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        if(k>nums.length)
        {
            return 0;
        }
        PriorityQueue<Integer> res=new PriorityQueue<>((a,b)->(b-a));
        for(int i=0;i<nums.length;i++)
        {
            res.offer(nums[i]);
        }
        int m=0;
        int count=0;
        while(count<k)
        {
            m=res.poll();
            count++;
        }
        return m;
    }
}

Solution2:

discussion的一种解法特别巧妙。用java自带的minheap,size小于k就加数,大于k就减掉最小的,这样子就能保证queue最顶端的就是最k大的。最后退出循环返回的就是第k大的数。

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        if(k>nums.length)
        {
            return 0;
        }
        PriorityQueue<Integer> res=new PriorityQueue<>();
        for(int i=0;i<nums.length;i++)
        {

            res.offer(nums[i]);
            if(res.size()>k)
            {
                res.poll();
            }
        }
        return res.poll();
    }
}

 

215. Kth Largest Element in an Array

原文:http://www.cnblogs.com/Machelsky/p/5904870.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!