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

Kth Largest Element in an Array

时间:2015-12-18 18:25:38      阅读:117      评论:0      收藏:0      [点我收藏+]

标签:

Kth Largest Element in an Array

Total Accepted: 32668 Total Submissions: 109273 Difficulty: Medium

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 {
private:
    int partion(vector<int>& nums,int start,int end){
    
        if(end-start==1){
            return start;
        }
    
        srand(time(0));
        int rindex = start + rand()%(end-start);
        swap(nums[rindex],nums[end-1]);
        
        int i=start-1,j=start;
        while(j < end-1){
            if(nums[j] < nums[end-1]){
                swap(nums[++i],nums[j]);
            }
            ++j;
        }
        
        swap(nums[++i],nums[end-1]);
        
        return i;
    }
public:
    int findKthLargest(vector<int>& nums, int k) {
        int start = 0;
        int end   = nums.size();
        
        k = end - k;
        
        int index = partion(nums,start,end);
        while(index != k){
            if(index < k){
                start = index+1;
            }else{
                end = index;
            }
            index = partion(nums,start,end);
        }
        return nums.size()==0 ? 0 : nums[k];
    }
};

 

Kth Largest Element in an Array

标签:

原文地址:http://www.cnblogs.com/zengzy/p/5057480.html

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