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

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

给定整数数组 nums 和整数 k，请返回数组中第 **k** 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0,right = nums.size()-1,target = nums.size()-k;
        while(left < right)
        {
            int mid = quick_selection(nums,left,right);

            if(mid == target) {
                return nums[mid];
            } else if(mid > target) {
                right = mid-1;
            } else {
                left = mid+1;
            } 
        }
        return nums[left];  // 如果只有一个元素；
    }
private:
    int quick_selection(vector<int>& nums,int left,int right) {
        int i=left+1,j=right;
        while(true) // 以nums[left]为基准循环比较
        {
            // 从左往右寻找第一个比nums[left]大的数的位置
            while(i<right && nums[i]<=nums[left])
            {
                ++i;
            }
            // 从右往左寻找第一个比nums[left]小的数的位置
            while(j>left && nums[j]>=nums[left])
            {
                --j;
            }
            if(j <= i)
                break;
            else
                swap(nums[i],nums[j]);
        }
        swap(nums[left],nums[j]);   // 交换
        return j;
    }
};