在未排序的数组中找到第 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
链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
基本思路: 快排
优化:因为只要第k大元素,在快排(排序成降序)中如果当前层次分好后,在partion的左边的数量多余K,就只要递归排序左半边就可以了。 这样每次就省去了一半的时间复杂度。
class Solution {
public:
int sort(vector<int>& nums, int l,int r,int& k){
if (l == r) return nums[l];
int i = l - 1, j = r + 1;
int x = nums[l];
while(i < j){
do ++i; while(nums[i] > x);
do --j; while(nums[j] < x);
if (i < j) swap(nums[i],nums[j]);
}
if (j + 1 >= k)
return sort(nums,l,j,k);
else
return sort(nums,j+1,r,k);
}
int findKthLargest(vector<int>& nums, int k) {
return sort(nums,0,nums.size() - 1,k);
}
};