Leetcode215-数组中的第K个最大元素

给定整数数组 nums 和整数 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次。最后堆顶元素即为所求。

class Solution {
public:
    void down(vector<int> & nums,int u,int size){
        int t=u;
        //nums的下标是从0开始的 所以 加一与加二
        if(u*2+1<size && nums[t]<nums[u*2+1]) t=u*2+1;
        if(u*2+2<size && nums[t]<nums[u*2+2]) t=u*2+2;
        if(t!=u){
            swap(nums[t],nums[u]);
            down(nums,t,size);
        }
    }
    int findKthLargest(vector<int>& nums, int k) {
        
        int size = nums.size();
        if(size==0||size==1) return nums[0];
        int n=size;
        for(int i=n/2;i>=0;i--) down(nums,i,n);
        int a=k-1;
        while(a--){
        	//注意：由于nums是由0开始，所以要特判size为1的情况。
            if(size!=1){
                nums[0]=nums[size-1];
                size--;
                down(nums,0,size);
            }else{
                nums[0]=nums[size];
                size--;
                down(nums,0,size);
            }
        }
        return nums[0];
    }
};