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;
    }
};
相关推荐
丶小鱼丶11 分钟前
栈算法之【有效括号】
java·算法
技术卷3 小时前
详解力扣高频 SQL 50 题之584. 寻找用户推荐人【入门】
sql·leetcode·oracle
小徐不徐说3 小时前
每日一算:华为-批萨分配问题
数据结构·c++·算法·leetcode·华为·动态规划·后端开发
菜鸟555556 小时前
图论:最小生成树
算法·图论
2401_872945097 小时前
【补题】Codeforces Round 735 (Div. 2) C. Mikasa
算法
叫我:松哥7 小时前
基于网络爬虫的在线医疗咨询数据爬取与医疗服务分析系统,技术采用django+朴素贝叶斯算法+boostrap+echart可视化
人工智能·爬虫·python·算法·django·数据可视化·朴素贝叶斯
Star在努力7 小时前
14-C语言:第14天笔记
c语言·笔记·算法
赴3358 小时前
Numpy 库 矩阵数学运算,点积,文件读取和保存等
人工智能·算法·numpy·random·dot
christ_lrs9 小时前
2025.7.25 测试 总结
贪心·模拟
自由随风飘9 小时前
机器学习-SVM支持向量机
算法·机器学习·支持向量机