牛客----寻找第K大(快速排序)

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

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

测试样例：[1,3,5,2,2],5,3

返回：2

 思路：快排+二分法
像快排一样，但是每次排完需要与k进行比较，具体比较看代码。
1.由于一般排序是从小到大排的；
所以注意这里比较的时候是 n-i 与 k进行比较；

代码如下：

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        return quick_sort(a,0,n-1,n,K);
    }
    
    int quick_sort(vector<int>& a,int start,int end,int n,int k){
        int base=a[start];
        int i=start;
        int j=end;
        int res=0;
        while(i<j){
            while(i<j&&a[j]>=base) --j;
            while(i<j&&a[i]<=base) ++i;
            if(i<j)
                swap(a[i],a[j]);
        }
        swap(a[start],a[j]);
        if(n-j==k)  return a[j];
        else if(n-j<k)  res=quick_sort(a,start,j-1,n,k);
        else if(n-j>k)  res=quick_sort(a,j+1,end,n,k);
       
        //下面i换成j也可以
        //swap(a[start],a[i]);
        //if(n-i==k)  return a[i];
        //else if(n-i<k)  res=quick_sort(a,start,i-1,n,k);
        // else if(n-i>k)  res=quick_sort(a,i+1,end,n,k);
        
        return res;
    }
    
};