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

【LeetCode 215】Kth Largest Element in an Array

时间:2015-06-16 14:43:22      阅读:161      评论: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.

 

思路:

       依据时间和空间复杂度,有多种方法。这里给出易于实现,基本能够满足面试、笔试要求的类似于快速排序的(递归)实现。

 

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        vector<int> leftvec;
        vector<int> rightvec;
        
        //以nums[0]作为分界元素,从nums[1]开始遍历
        //比nums[0]小的元素放入leftvec,反之放入rightvec
        for(int i = 1; i < nums.size(); i++)
        {
            if(nums[i] <= nums[0])
                leftvec.push_back(nums[i]);
            else
                rightvec.push_back(nums[i]);
        }
        
        //rightvec的长度即为大于nums[0]的元素的个数
        int len = rightvec.size();
        
        if(len >= k)//前k大的元素都在rightvec中,递归求解rightvec
        {
            return findKthLargest(rightvec, k);
        }
        else if(len == k - 1)//比nums[0]大的元素有k-1个,那么nums[0]即为所求结果
        {                            
            return nums[0];
        }
        else//比nums[0]大的元素有[0, k-2]个,说明第K大的元素在leftvec中,递归求解leftvec               
        {    
            return findKthLargest(leftvec, k - len - 1);
        }
    }
};    

 

【LeetCode 215】Kth Largest Element in an Array

标签:

原文地址:http://www.cnblogs.com/tjuloading/p/4580615.html

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