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

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

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array

基本思路: 快排

优化：因为只要第k大元素,在快排(排序成降序)中如果当前层次分好后,在partion的左边的数量多余K，就只要递归排序左半边就可以了。 这样每次就省去了一半的时间复杂度。

class Solution {
public:
    
    int sort(vector<int>& nums, int l,int r,int& k){
        if (l == r) return nums[l];
        int i = l - 1, j = r + 1;
        int x = nums[l];
        
        while(i < j){
            do ++i; while(nums[i] > x);
            do --j; while(nums[j] < x);
            if (i < j) swap(nums[i],nums[j]);
        }
        if (j + 1 >= k)
            return sort(nums,l,j,k);
        else
            return sort(nums,j+1,r,k);
    }
    
    int findKthLargest(vector<int>& nums, int k) {
       return sort(nums,0,nums.size() - 1,k);
    }
};