[算法]parition求TopK

parition算法是用于解决TopK问题代价最小的算法之一
时间O(n)  空间O(1)


算法流程:

  1. 引入随机化选取一个枢轴并与尾元素交换
  2. 将区间内元素排为小于枢轴在前,大于枢轴在后的形式 使得i-lst大于等于枢轴
  3. 比较i与k大小递归处理
  4. 返回元素
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int fst = 0, lst = nums.size() - 1;
        k = lst - k + 1;
        while(fst <= lst)
        {
            int pos = fst + ( rand() % (lst - fst + 1) );
            swap(nums[lst], nums[pos]);
            int i = fst, j = fst;
            while(j < lst)
            {
                if(nums[j] <= nums[lst])
                    swap(nums[i++], nums[j]);
                ++j;
            }
            swap(nums[i], nums[j]);
            if(i == k)
            {
                break;
            }
            else if(i > k)
            {
                lst = i - 1;
            }
            else
            {
                fst = i + 1;
            }
        }
    return nums[k];
    }
};

 

 

 

 

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值