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];
}
};