面试必考精华版Leetcode2462. 雇佣 K 位工人的总代价

meeiuliuus

于 2023-12-05 10:01:00 发布

阅读量47

点赞数

分类专栏： # leetcode ---medium 文章标签：前端算法 javascript

本文链接：https://blog.csdn.net/qq_52313711/article/details/134799808

版权

leetcode ---medium 专栏收录该内容

174 篇文章

订阅专栏

本文介绍了一段C++代码，通过优先队列实现计算在给定成本向量中，满足特定条件下的最小总成本。算法涉及到两个优先队列，分别从两端处理，直至满足k次操作。

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

题目：

代码（首刷看解析）：

class Solution {
public:
    long long totalCost(vector<int>& costs, int k, int candidates) {
        // 优先队列，升序
        priority_queue<int, vector<int>, greater<int>> frontQue;
        priority_queue<int, vector<int>, greater<int>> backQue;
        int n = costs.size();
        for(int i=0;i<candidates;++i) frontQue.push(costs[i]);
        for(int i=n-1;i>=n-candidates&&i>=candidates;--i) backQue.push(costs[i]);//保证n-canditates不是负数
        int lcursor=candidates,rcursor=n-candidates-1;
        long long sum = 0;
        while(k){
            long long lMin=INT_MAX,rMin=INT_MAX;
            if(!frontQue.empty()) lMin = frontQue.top();
            if(!backQue.empty()) rMin = backQue.top();
            if(lMin<=rMin){
                frontQue.pop();
                if(lcursor<=rcursor) frontQue.push(costs[lcursor++]);
                sum+=lMin;
            }else{
                backQue.pop();
                if(rcursor>=lcursor) backQue.push(costs[rcursor--]);
                sum+=rMin;
            }
            k--;
        }
        return sum;
    }
};