LeetCode Kth Largest Element in an Array （快速排序）

 

 

题意：

　　在一个无序的数组中第k大的数是多少？

 

思路：

　　按照快排的思路，如果每次分成两段后，设为L和R。如果R>=k ，则答案在右边集合，否则在左边集合。

　　这里用了3位取中法。注意快排别给写死循环了。

 

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        if(k>nums.size())    return 0;
           return DFS(nums,0,nums.size()-1,k);
    }

    int DFS(vector<int>& nums,int s,int e,int k)
    {
        int L=s, R=e;
        //三位取中法
        if(nums[e]>nums[s])    swap(nums[s],nums[e]);    
        if(nums[s]>nums[(s+e)/2])    swap(nums[s],nums[(s+e)/2]); 

        int mid=nums[s];
        while(L<R)
        {
            while(L<R && nums[R]>=mid)    R--;    //找小
            nums[L]=nums[R];
            while(L<R && nums[L]<=mid)    L++;    //找大
            nums[R]=nums[L];
        }
        nums[L]=mid;
        int len=e-L;//右边部分的元素个数
        if(len+1==k)    return mid;
        if(len>=k)    return DFS(nums,L+1,e,k);
        else        return DFS(nums,s,L-1,k-len-1);
    }
};