快速选择+最小堆 215. 数组中的第K个最大元素

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

说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

最小堆

创建一个最大堆，保留k个数，遍历一次后取出堆顶即可；

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int,vector<int>,greater<int>> q;
        for(int i=0;i<nums.size();i++)
        {   
            q.push(nums[i]);
            if(q.size()>k) q.pop();
        }
        return q.top();
    }
};

快速选择

利用快速排序中的快速选择；
随机取一个数进行partition；

partition
（1）将pivot数放到该次选择的末尾right位置；
（2）从left遍历到right，将比pivot大的数放在前面；
（3）完成后将pivot换回到当前待放置的位置；
返回该下标！

findk
（1）在left和right范围内选取一个pivot下标，传入partition函数；
（2）若partition返回的下标为第k个数，返回该数；
（3）若left==right，返回当前剩余的最后一个数；（省一次函数调用）；
（4）若返回的p在第k个位置前面，递归遍历后半部分p+1~right；
（5）若返回的p在第k个位置后面，递归遍历前半部分left~p-1；

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        return findk(nums,k,0,n-1);
    }
private:
    int findk(vector<int> & nums,int k,int left,int right){
        if(left==right) return nums[left];
        int p=partition(nums,left,left,right);
        if(p==k-1) return nums[p];
        else if(p<k-1) return findk(nums,k,p+1,right);  //太大了，找右边的
        else return findk(nums,k,left,p-1);             //太小了，找左边的
    }
    int partition(vector<int>& nums, int pivot_index,int left,int right)        //第k小的 在left和right中找
    {
        //把pivot放到最右边
        int start_index=left;
        int pivot=nums[pivot_index];
        swap(nums[pivot_index],nums[right]);

        //比pivot大的放在最左边
        for(int i=left;i<=right;i++){  
            if(nums[i]>pivot){        //相同元素也算
                swap(nums[i],nums[start_index]);
                start_index++;
            }
        }
        //pivot归位
        swap(nums[start_index],nums[right]);
        return start_index;      //返回pivot的位置
    }
};