159.库存管理(TOPk问题!)

本文介绍了两种方法实现库存管理,一种是基于快速排序的分块思想,另一种是利用优先级队列。两种算法都旨在找到并返回前k个最小的元素。

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

思路:也是tok的问题,与上篇博客思路一样,只不过是求前k个小的元素!

基于快排分块思路的代码如下:

class Solution {
public:
    int getkey(vector<int>&nums,int left,int right)
    {
        int r=rand();
        return nums[r%(right-left+1)+left];
    }
    void qsort(vector<int>&nums,int left,int right,int k)
    {
        if(left>=right ) return ;
        vector<int>ret(k);
        int l=-1,r=right+1;
        int i=0;
        int key=getkey(nums,left,right);
        while(i<r)
        {
            if(nums[i]<key)
            {
                swap(nums[++l],nums[i++]);
            }
            else if(nums[i]>key)
            {
                swap(nums[--r],nums[i]);
            }
            else
            {
                i++;
            }
        }

        int a=l-left+1;
        int b=r-l-1;
        int c=right-r+1;

        if(a>=k)
        {
            return qsort(nums,left,l,k);
        }
        else if(a+b>=k)
        {
            return ;
        }
        else
        {
            return qsort(nums,r,right,k-a-b);
        }
        return ;
    }

    vector<int> inventoryManagement(vector<int>& stock, int cnt) 
    {
        int n=stock.size();
        qsort(stock,0,n-1,cnt);
        return {stock.begin(),stock.begin()+cnt};
    }
};

 优先级队列代码:

class Solution {
public:
    vector<int> inventoryManagement(vector<int>& stock, int cnt)
    {
        vector<int>ret(cnt);
        priority_queue<int,vector<int>,greater<int>>q;
        for(auto ch:stock)
        {
            q.push(ch);
        }
        for(int i=0;i<cnt;i++)
        {
            ret[i]=q.top();
            q.pop();
        }       
        return ret;
    }
};

评论 27
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值