本题是快速排序的变形题目。题目描述如下:
本题的基本思想如下:
在快排每次划分区间的过程中,计算比枢轴元素大的元素个数sr,即sr = nums.size()-j-1. 然后将sr与k作对比:若k<=sr,则取右半边进行递归;否则取左半边进行递归。注意边界条件是left==right,此时返回nums[left].
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int left = 0, right = nums.size()-1;
int res = partition(nums,left,right,k);
return res;
}
int partition(vector<int>&nums,int left, int right,int k){
if(left==right) return nums[left];
int i=left-1, j=right+1, x=nums[left];
while(i<j){
do i++; while(nums[i]<x);
do j--; while(nums[j]>x);
if(i<j) swap(nums[i],nums[j]);
}
int sr = nums.size()-j-1;
if(k<=sr) return partition(nums,j+1,right,k);
else return partition(nums,left,j,k);
}
};
本算法的平均时间复杂度是O(n),而如果简单地使用快速排序的平均时间复杂度为O(nlogn)