【Leetcode】857. 雇佣 K 名工人的最低成本

857. 雇佣 K 名工人的最低成本

有 n 名工人。 给定两个数组 quality 和 wage ，其中quality[i] 表示第 i 名工人的工作质量，其最低期望工资为 wage[i] 。

现在我们想雇佣 k 名工人组成一个*工资组。*在雇佣 一组 k 名工人时，我们必须按照下述规则向他们支付工资：

1. 对工资组中的每名工人，应当按其工作质量与同组其他工人的工作质量的比例来支付工资。
2. 工资组中的每名工人至少应当得到他们的最低期望工资。

给定整数 k ，返回 组成满足上述条件的付费群体所需的最小金额 。在实际答案的 10-5 以内的答案将被接受。

示例 1：

输入： quality = [10,20,5], wage = [70,50,30], k = 2
输出： 105.00000
解释： 我们向 0 号工人支付 70，向 2 号工人支付 35。

示例 2：

输入： quality = [3,1,10,10,1], wage = [4,8,2,2,7], k = 3
输出： 30.66667
解释： 我们向 0 号工人支付 4，向 2 号和 3 号分别支付 13.33333。

提示：

• n == quality.length == wage.length
• 1 <= k <= n <= 104
• 1 <= quality[i], wage[i] <= 104

---

解题思路

题目意思的要求就是找到一种满足条件的最少工资方案，题目给出了两个条件。

1. 所有人按照工作质量的比例来支付工资。
2. 每名工人的工资至少是他们期望的最低工资。

可以从两个条件中挖掘出那些信息呢？

1️⃣从第一个条件来看，所有人都是按照工作质量的比例来发工资，所有总得有一个基准工资吧，就是单个质量的工资多少钱，所以我们只需要找到这个基准工资，并且在满足条件2的情况下去寻找到k个工人，然后再去寻找最小的总工资。

大体方向有了，找基准工资，我们可以把所有人的基准工资求出来，再去一一找k个工人。但是这样可能会导致不满足条件二的情况，就是如果选了单个质量工资低的人作为基准，去乘以其他人的工作质量求出来的工资可能会小于此工人的最低期望工资的。所以需要在满足条件二的情况下去寻找基准工资， 如何来寻找呢？

2️⃣可以对所有工人的单个质量的工资排升序，工资从下标为k-1的工人开始向后遍历，任意选择前面k个工人(包括本身)。

提问：为什么从第k-1开始遍历，而不从0开始遍历？

因为从k-1个开始向后遍历，一定可以使得前面k个工人得到薪资一定是大于最低期望薪资，因为是排升序，所以后面人的单个质量薪资是大于前面工人的，所以乘以工作质量所得到的工资一定是超过最低期望薪资的，所以就完美解决了满足条件2的问题。

3️⃣接下来就是寻找最小的总薪资的问题了，因为基准工资是固定的，所以只需要寻找到对应的k个工人的最小的工作质量和就可以了，相乘就是此基准工资条件下的最少总工资。

既然是k个工人，工人数是固定的，所以可以想到队列或者优先队列(堆)来做。本题用优先队列维护k个数的和最小，所以可以使用大根堆，遇到比堆顶小的元素可以直接将堆顶删除，将较小元素加入堆。同时基准工资向后走，求出满足条件二的所有基准工资下的总的工资数，找到最小值。

---

代码实现

class Solution
{
public:
    double mincostToHireWorkers(vector<int> &quality, vector<int> &wage, int k)
    {
        vector<pair<double, int>> v; // 使用pair，因为在排序的时候也需要保存下来下标
        //求出每个工人的单位质量的薪资存储到vector中
        for (int i = 0; i < quality.size(); i++)
            v.push_back(make_pair((double)wage[i] / quality[i], i));//注意强转为double类型
        //按照单位质量的薪资排升序
        sort(v.begin(), v.end()); 
        
        priority_queue<int> q; // 大根堆
        int sumq = 0; //记录堆中k个元素的总和
        //将初始的k个工人的工作质量全部放入堆中
        for (int i = 0; i < k; i++)
        {
            q.push(quality[v[i].second]);
            sumq += quality[v[i].second];//sumq维护堆中k个元素的总和
        }
        //计算出起始的工人总工资
        double res = v[k - 1].first * sumq;
        
        //从下标k开始遍历，因为k-1的情况已经储存在堆中了
        for (int i = k; i < v.size(); i++)
        {
            //如果发现存在工作质量小于堆顶元素，就更新堆
            if (quality[v[i].second] < q.top())
            {
                sumq -= q.top() - quality[v[i].second];//更新堆中元素总和
                q.pop(); //删除堆顶
                q.push(quality[v[i].second]);//加入较小元素
                res = min(res, sumq * v[i].first);//更新工人的总的工资res
            } 
            //只有sumq变小了，res才有可能变小，因为后面的基准工资大于前面的，所以如果sumq没变化，工人总工资一定是变大的
        }
        return res;
    }
};