Leetcode 857 -- 贪心 | 数学

题目描述

雇佣 K 名工人的最低成本

思路

参考官方题解和这里

代码1(正确)

复制代码
class Solution {
public:
    double mincostToHireWorkers(vector<int>& quality, vector<int>& wage, int k) {
        int n = wage.size();
        double res = 0, totalq = 0; // totalquality
        vector<int> h(n);
        for(int i = 0; i < n; i ++ )    h[i] = i;
        sort(h.begin(), h.end(), [&](int &a, int &b){
           return wage[a] * quality[b] < wage[b] * quality[a]; 
        });
        for(int i = 0; i < n; i ++ )    cout << h[i] << ' ';
        cout << endl;
        priority_queue<int, vector<int>, less<int>> q; // 降序
        for(int i = 0; i < k - 1; i ++ )
        {
            q.push(quality[h[i]]);
            totalq += quality[h[i]];
        }
        for(int i = k - 1; i < n; i ++ )
        {
            q.push(quality[h[i]]);
            totalq += quality[h[i]];
            double cur = totalq / (double)quality[h[i]] * wage[h[i]];
            if(res == 0 || cur < res)  res = cur;
            totalq -= q.top();
            q.pop();
        }
        
        return res;
    }
};

代码2(错误)

复制代码
class Solution {
public:
    typedef pair<double, int> PDI;
    double mincostToHireWorkers(vector<int>& quality, vector<int>& wage, int k) {
        int n = wage.size();
        double res = 0, totalq = 0; // totalquality
        vector<int> h(n);
        for(int i = 0; i < n; i ++ )    h[i] = i;
        sort(h.begin(), h.end(), [&](int &a, int &b){
           return quality[a] < quality[b];
        });
        
        priority_queue<PDI, vector<PDI>, less<PDI>> q; // 降序
        for(int i = 0; i < k - 1; i ++ )
        {
            int idx = h[i];
            totalq += quality[idx];
            q.push({(double)wage[i] / quality[i],idx});
        }
        for(int i = k - 1; i < n; i ++ )
        {
            int idx = h[i];
            q.push({(double)wage[idx] / quality[idx],idx});          
            totalq += quality[idx];
            auto t = q.top();
            cout << totalq << ' ';
            double cur = totalq * t.first;
            if(res == 0 || cur < res)   res = cur;
            totalq -= quality[t.second];
            q.pop();
        }
        
        return res;
    }
};
相关推荐
雾月5528 分钟前
LeetCode 1292 元素和小于等于阈值的正方形的最大边长
java·数据结构·算法·leetcode·职场和发展
OpenC++1 小时前
【C++QT】Buttons 按钮控件详解
c++·经验分享·qt·leetcode·microsoft
知来者逆2 小时前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊2 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北2 小时前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈3 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing3 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD4 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘5 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV5 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法