leetcode——215.数组中的第K个大元素

Your_Julia

于 2021-06-17 16:42:51 发布

阅读量94

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/Dedication_/article/details/117996598

Leetcode 专栏收录该内容

145 篇文章

订阅专栏

本文介绍了一种利用快速排序的partition过程来高效寻找数组中第K大元素的方法。通过随机选取枢轴并进行分区操作，可以有效减少搜索范围，实现O(n)平均时间复杂度的查找。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

利用快排的partition确定第k大的位置在哪
快排的递归条件为三个是否找到，没有找到接下来往左递归还是往右递归?
为了加快算法速度，引入随机种子，每次挑选不一样的中枢

代码：

class Solution {
public:
    //根据返回的q与index的关系，决定是否继续递归，向哪边递归
    int quickSelect(vector<int>& a, int l, int r, int index) {
        int q = partition(a, l, r);
        if (q == index) 
            return a[q];
        else
            return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1,index);
    }
    //快排的Partition
    inline int partition(vector<int>& a, int l, int r) {
    	//引入随机种子，加快算法效率
        srand(time(0));
        int random = rand() % (r - l + 1) + l;
        swap(a[random], a[l]);
        //正常快排的partition步骤
        int x = a[l];
        while(l<r){
            while(l<r&&a[r]>=x) r--;
            a[l]=a[r];
            while(l<r&&a[l]<=x) l++;
            a[r]=a[l];
        }
        a[l]=x;
        return l;
    }
    //主函数 第k大的元素在数组中的index为nums.size()-k;
    int findKthLargest(vector<int>& nums, int k) {
        return quickSelect(nums, 0, nums.size() - 1, nums.size() - k);
    }
};