寻找第K大

最新推荐文章于 2022-03-25 09:27:21 发布

vector6_

最新推荐文章于 2022-03-25 09:27:21 发布

阅读量766

点赞数 1

分类专栏： Leetcode 文章标签：数据结构算法 leetcode

本文链接：https://blog.csdn.net/qq_41105501/article/details/116885497

版权

Leetcode 专栏收录该内容

137 篇文章 0 订阅

订阅专栏

这篇博客介绍了两种方法来寻找整数数组中的第K大元素：一种是利用优先队列（大顶堆），另一种是采用快速选择算法。优先队列方法将数组元素按大小顺序插入堆中，然后弹出K个最小的元素，最后堆顶的元素即为第K大的数。快速选择算法则是通过类似于快速排序的分割过程，递归地在左右子数组中寻找目标位置。这两种方法都能有效解决该问题。

摘要由CSDN通过智能技术生成

寻找第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);
    }
};