您的位置:首页 > 其它

LeetCode 215. Kth Largest Element in an Array

2017-03-05 21:12 351 查看
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.
如题所示,求未排序的数组中最大的元素。
第一个想法是使用快排,快排的思想是“挖坑”和“分治”,主要有以下步骤:
1.选定一个位置(如初始位置),将该位置的数“挖”出来,保存到临时变量temp中,这样就留下了第一个坑。
2.从后往前找,看是否有小于temp的数,如果有就将其放入之前的坑中,此时留下了第二个坑。
3.再从前往后找,看是否有大于temp的数,如果有就将其放入第二个坑中。
接下来重复步骤2和步骤3,直到左右两边的下标相遇。
4.再将temp放入下标中。
这样就得到了以初始的数为中心,左边比它小,右边比它大的序列。
5.最后,只需要对L~i-1和i+1~R两部分递归就行。
由于我们是寻找第K大的数,因此只需要判断这个数是属于L~i-1还是i+1~L,因此只需要对一半的数字进行排序,这样就能大大地简化步骤。

代码实现如下:
void quick_sort(vector<int> &nums,int l,int r,int k){
int i=l,j=r;
if(i<j){
int temp = nums[l];
while(i < j){
while(i < j && temp <= nums[j])
j--;
if(i<j){
nums[i] = nums[j];
i++;
}
while(i<j &&temp >= nums[i])
i++;
if(i<j){
nums[j] = nums[i];
j--;
}
}
nums[i] = temp;
if(i == nums.size()-k)
return ;
else if(i<nums.size()-k)
quick_sort(nums,i+1,r,k);
else
quick_sort(nums,l,i-1,k);
}

}

class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
quick_sort(nums,0,nums.size()-1,k);
return nums[nums.size()-k];
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode K-th Largest