leetcode 215 第k大元素

最新推荐文章于 2025-05-11 08:18:33 发布

原创最新推荐文章于 2025-05-11 08:18:33 发布 · 119 阅读

0 ·

CC 4.0 BY-SA版权

双指针同时被 3 个专栏收录

18 篇文章

订阅专栏

排序

8 篇文章

订阅专栏

堆

7 篇文章

订阅专栏

本文介绍了一种使用快速选择算法寻找数组中第K大的元素的方法。该算法基于快速排序，通过调整递归方式实现高效查找。文章详细解释了快速选择算法的具体实现步骤，并提供了一个C++代码示例。

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

解法一：维度大小为k的堆
复杂度: $O (l N o g K)$

解法二：快速选择
快排其实还是有挺多细节的，
1 必须是右边的哨兵先走，因为要保证最后停在偏小的位置上，然后让哨兵和他交换位置。
2 递归终止条件可以设置为 l >= r，可以保证一个元素，或者出错时能停止。

class Solution {
public:
    void qChoose(vector<int> &nums, int k, int l, int r)
    {
        if(l >= r)return;

        int pivot = l, target = nums[l];
        int pl = l, pr = r;
        while(pl < pr)
        {
            while(pl < pr && nums[pr] >= target)pr--;
            while(pl < pr && nums[pl] <= target)pl++;

            if(pl < pr)swap(nums[pl], nums[pr]);
        }
        swap(nums[l], nums[pl]);

        if(pl == k)return;

        if(pl < k)qChoose(nums, k, pl+1, r);
        else qChoose(nums, k, l, pl-1);
    }

    int findKthLargest(vector<int>& nums, int k) {
        int n = nums.size();
        qChoose(nums, n-k, 0, n-1);
        return nums[n-k];
    }
};