寻找第K大——牛客NC88

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

1.快排

class Solution {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        int res=quickSort(a,0,n-1,K,n);
        return res;
    }
    int quickSort(vector<int> &a,int left,int right,int k,int n){
        int i=left;
        int j=right;
        int res=0;
        int temp=a[left];
        while(i!=j){
            while(a[j]>=temp&&i<j)
                j--;
            while(a[i]<=temp&&i<j)
                i++;
            if(i<j){
                swap(a[i],a[j]);
            }
        }
        a[left]=a[i];
        a[i]=temp;
        if(n-i==k)
            return a[i];
        else if(k>n-i){
            res=quickSort(a,left,i-1,k,n);
        }else if(k<n-i){
            res=quickSort(a,i+1,right,k,n);
        }
        return res;
    }
   
};

2.堆排序

class Solution {
public:
    int findKth(vector<int> &nums, int n, int k) {
        // write code here
        int length=nums.size();
        bulidHeap(nums,length);
        for(int i=0;i<k-1;i++){
            int re=nums[length-1];
            nums[length-1]=nums[0];
            nums[0]=re;
            length--;
            bulidHeapTool(nums,0,length);
        }
        return nums[0];
    }
    vector<int>& bulidHeapTool(vector<int>& nums,int i,int length){
        int left=2*i+1,right=2*i+2;
        if(right>=length){
            if(left>=length) return nums;
            else if(nums[left]>nums[i]){
                int re =nums[i];
                nums[i]=nums[left];
                nums[left]=re;
                return nums;
            }
            else return nums;
        }
        int max=(nums[left]>nums[right])?left:right;
        if(nums[max]<=nums[i]) return nums;
        else{
            int re=nums[i];
            nums[i]=nums[max];
            nums[max]=re;
        }
        return bulidHeapTool(nums,max,length);
    }
    vector<int>& bulidHeap(vector<int>& nums,int length)
    {
        int i=(length)/2-1;
        for (i;i>=0;i--){
            nums=bulidHeapTool(nums,i,length);
        }
        return nums;
    }
   
};