题目地址(857. 雇佣 K 名工人的最低成本)

题目地址(857. 雇佣 K 名工人的最低成本)

https://leetcode.cn/problems/minimum-cost-to-hire-k-workers/

题目描述

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

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

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

给定整数 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

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

class Solution:
    def mincostToHireWorkers(self, quality: List[int], wage: List[int], k: int) -> float:
        # 采用大根堆
        efforts = [(q/w,q) for q,w in zip(quality,wage)]
        # 按照工作效率q/w，从大到小进行排列
        # 采用固定大小的大根堆，那么就可以保证就取到了尽可能小的total，也保证了全体采用最低工作效率
        efforts.sort(key=lambda a:-a[0])
        
        ans = float('inf') 
        total = 0 
        h = []
        for eff, q in efforts:
            total += q
            heapq.heappush(h,-q)
            # 这里需要先判断len(k)>k
            # k+1个人中，确定了最低的工作效率，然后确定最低的工作质量总和
            if len(h)>k:
                total += heapq.heappop(h)
            if len(h) == k:
                ans = min(ans,total/eff)
        return ans 
            

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$