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

215. Kth Largest Element in an Array

时间:2016-09-25 07:24:17      阅读:136      评论: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
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!