215. 数组中的第K个最大元素

题目描述

点这里

思路分析

快速排序的经典应用
可以递归改循环优化节省掉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;;
        }
    }
};