题目描述
思路分析
快速排序的经典应用
可以递归改循环优化节省掉O(logn)的空间
代码实现
//朴素快速选择,O(n)时间O(logn)空间
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
//srand(time(0)); // 随机种子
return dfs(nums,0,nums.size()-1,k);
}
int dfs(vector<int> &nums,int l,int r,int k){
if(l>=r) return nums[l];
int dex=nums[l+r>>1],i=l-1,j=r+1;
//dex=nums[rand() % (r - l + 1) + l];
while(i<j){
do i++;while(nums[i]>dex);
do j--;while(nums[j]<dex);
if(i<j){
int t=nums[i];
nums[i]=nums[j];
nums[j]=t;
}
}
if (j-l+1>=k) return dfs(nums, l, j, k);
else return dfs(nums, j + 1, r, k - (j - l + 1));
}
};
//优化快速选择,O(n)时间O(1)空间
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
//srand(time(0));
return dfs(nums,0,nums.size()-1,k);
}
int dfs(vector<int> &nums,int l,int r,int k){
while(1){
if(l>=r) return nums[l];
int dex=nums[l+r>>1],i=l-1,j=r+1;
//dex=nums[rand() % (r - l + 1) + l];
while(i<j){
do i++;while(nums[i]>dex);
do j--;while(nums[j]<dex);
if(i<j){
int t=nums[i];
nums[i]=nums[j];
nums[j]=t;
}
}
if (j-l+1>=k) r=j;
else k = k - (j - l + 1), l = j + 1;;
}
}
};