在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
方法1:
快排:
这里采用两路快排。
1、输入数组,进入快排;
2、需要对哪一个数组进行排序,范围是多少,进入快排子函数;
3、选取判断的标定点,标定点前得数小于标定点,标定点后面的数大于等于标定点。然后分别再对标定点前、后的数组进行排序,迭代操作,直至不能再迭代。
4、标定点确定函数:[l+1,j]<v 初始化j =l;
标定点需要做随机选择,避免算法退化,随机从范围内选择(r-l+1)+l (区间+偏移量)。
选择好标定点后,开始从l+1遍历,小于标定点的,j++;然后交换nums[i]和nums[j],否则,不进行操作。
最后将nums[i]和nums[j]交换,返回标定点j。
代码如下:
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
quicksort(nums,nums.size());
return nums[nums.size()-k];
}
int __partition(vector<int>& arr, int l, int r){
swap( arr[l] , arr[rand()%(r-l+1)+l] );
int v = arr[l];
int j = l;
for(int i =l+1; i <= r; i++){
if(arr[i] < v){
j++;
swap(arr[j], arr[i]);
}
}
swap(arr[l],arr[j]);
return j;
}
void __quicksort(vector<int>& arr, int l, int r){
if(l>=r) return;
int p = __partition(arr, l,r);
__quicksort(arr,l,p-1);
__quicksort(arr,p+1,r);
}
void quicksort(vector<int>& arr, int n ){
__quicksort(arr, 0 ,n-1);
}
};