总体思路:利用快速排序,但是每次只需要对k在的那一半排队即可,而不是像一般快排两部分都要排序。所以叫做快速选择。
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
return quickSort(nums, 0, nums.size() - 1, nums.size() - k);
}
int quickSort(vector<int>& nums, int l, int r, int k) {
if (l >= r) return nums[k];
int x = nums[l], i = l - 1, j = r + 1;
while (i < j) { // 快排分配区间,将小于x的放到一边,大于的在另一边。具体通过两个指针向中间逼近,当所指都不满足时交换两个数值,则又满足了。
do i++; while (nums[i] < x);
do j--; while (nums[j] > x);
if (i < j) swap(nums, i, j);
}
if (k <= j) return quickSort(nums, l, j, k);
else return quickSort(nums, j + 1, r, k);
}
void swap(vector<int>& nums, int i, int j) {
int c = nums[i];
nums[i] = nums[j];
nums[j] = c;
}
};