159.库存管理（TOPk问题！）

原创已于 2023-12-02 23:00:24 修改 · 987 阅读

37 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #数据结构 #学习方法 #c++

于 2023-12-02 22:59:51 首次发布

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

思路：也是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;
    }
};