topK问题

下面代码:求解第k大的数
力扣215:
https://leetcode-cn.com/problems/kth-largest-element-in-an-array/

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int len = nums.size();
        int left = 0;
        int right = len - 1;
        int target = len - k;
        while(true)
        {
            int index = partition(nums, left, right);
            if(target == index)
            {
                return nums[index];
            }
            else if(target > index)
            {
                left = index + 1;
            }
            else
            {
                right = index - 1;
            }
        }

    }
    int partition(vector<int>& nums, int left, int right)
    {
        int tmp = nums[left];
        while(left < right)
        {
            while(left < right && nums[right] > tmp)
                right--;
            nums[left] = nums[right];
            while(left < right && nums[left] <= tmp)
                left++;
            nums[right] = nums[left];
        }
        nums[left] = tmp;
        return left;
    }
};
//  代码参考  https://leetcode-cn.com/problems/kth-largest-element-in-an-array/solution/partitionfen-er-zhi-zhi-you-xian-dui-lie-java-dai-/

//  讲解参考  https://www.bilibili.com/video/BV16i4y1x7TN?from=search&seid=6656008681955547018

上面给出的代码都是求解第k大元素;若想要得到Top K元素(前k大),仅需要将代码做稍微的修改:比如,扫描完成后的小顶堆对应于Top K,Quick Select算法用中间变量保存Top K元素。

文中最后一段话:https://www.cnblogs.com/qlky/p/7512199.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

触不可及<>

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值