寻找第K大

Posted 2021-05-20 vector6_

寻找第K大

有一个整数数组，找出数组中第K大的数。

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

class Solution {
public:
    //使用优先队列（默认大顶堆）
//     int findKth(vector<int> a, int n, int K) {
//         priority_queue<int,vector<int>,greater<int>> pqueue; //默认大顶堆,使用小顶堆需要使用 greater<int>
//         int index = 0;
//         while(index < K)
//         {
//             pqueue.push(a[index]);
//             ++index;
//         }
//         while(index<n)
//         {
//             int minValue = pqueue.top();
//             if(a[index]>minValue)
//             {
//                 pqueue.pop();
//                 pqueue.push(a[index]);
//             }
//             ++index;
//         }
//         return pqueue.top();
//     }
    
    //使用快排的思想，先分割，若右侧数字个数不够K个，则从左侧找第k-bigNum-1大的数，如果右侧数字个数比K多，则在右侧找第K大的数
    int quickFind(vector<int>& a, int left, int right, int K)
    {
        int i = left, j = right;
        int mark = a[left];
        while(i < j)
        {
            while(i<j && a[j] >= mark)
                --j;
            a[i] = a[j];
            while(i<j && a[i] <= mark)
                ++i;
            a[j] = a[i];
        }
        a[i] = mark;
        
        int bigNum = right - i;
        if(bigNum == K-1)
            return mark;
        else if(bigNum < K-1)
        {
            K = K-1-bigNum;
            return quickFind(a, left, i-1, K);
        }
        else
        {
            return quickFind(a, i+1, right, K);
        }
    }
    
    int findKth(vector<int> a, int n, int K)
    {
        return quickFind(a, 0, n-1, K);
    }
};