LeetCode215 数组中的第k最大元素

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

 

直接用快速排序的partition函数，可以在O(n)时间内解决。或者使用最大堆，时间复杂度为O(nlogn)。

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

    int partition(vector<int>& nums,int left, int right, int k){
        if(left==right)
            return nums[left];
        int standard=nums[left];
        int orileft=left,oriright=right;
        while(left<right){
            while(left<right && nums[right]>=standard)
                --right;
            if(left<right)
                nums[left]=nums[right];
            
            while(left<right && nums[left]<standard)
                ++left;
            if(left<right)
                nums[right]=nums[left];
        }
        nums[left]=standard;
        cout<<nums[left]<<endl<<left<<endl;
        if(left==k)
            return standard;
        else if(left>k)
            return partition(nums,orileft,left-1,k);
        else 
            return partition(nums,left+1,oriright,k);
    }
};