题目链接与描述
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
关键词:贪心算法 优先级队列
对工资组中的每名工人,应当按其工作质量与同组其他工人的工作质量的比例来支付工资。
工资组中的每名工人至少应当得到他们的最低期望工资。
方法一:
运行截图
代码
public double mincostToHireWorkers(int[] quality, int[] wage, int k) {
// 对工资组中的每名工人,应当按其工作质量与同组其他工人的工作质量的比例来支付工资。
//工资组中的每名工人至少应当得到他们的最低期望工资。
// 首先我们得用公式推导:
// k个人:1-k ,某个人是i
// 对应的质量:q[i],对应的工资期待是w[i]
// 实际的收入是和质量成比例的,也就是假定比例是t[i] = sum(q[1]...q[k])/k
// 那么实际的收入就比例乘自身质量:r[i] = t[i]*q[i],题目中,必须不低于最低期望,所以r[i] >= w[i]
// 综合可以得到 t[i]*q[i]>=w[i] 等价于 t[i] >= w[i]/q[i]
// 而题目是为了 找k个人,给最少的钱,质量不看重;所以只需要找比例 t[i] 最大的k个人就好了
// 第一步要算出k个人,每人的质量比就好了 w[i]/q[i] > w[j]/q[j]
// 所以才会有w[i]*q[j]-w[j]*q[i] 的排序,还可以避免精度问题
Integer[] sort = IntStream.range(0, quality.length).boxed().toArray(Integer[]::new);
Arrays.sort(sort, (l, r) -> quality[r] * wage[l] - quality[l] * wage[r]);
// 得到了质量比的排序后,有三个变量,质量、花费、最低期望:所以锁定第k人质量最高情况,k-1人质量越低,花费工资越少,然后再对比第k人的质量情况
// 总的结果
double res = 1e9;
// 总的质量
double totalq = 0.0;
// 创建优先级队列
PriorityQueue<Integer> pq = new PriorityQueue<Integer>((a, b) -> b - a);
// 取k个人的质量总和
for (int i = 0; i < sort.length; i++) {
int idx = sort[i];
// 取前k-1个人 性价比高的 得到质量总和
totalq += quality[idx];
// 将当前质量加入优先级队列,质量低的优先
pq.offer(quality[idx]);
if (i >= k - 1) {
// 剩余一个对比花费最少
res = Math.min(res, ((double) wage[idx] / quality[idx]) * totalq);
totalq -= pq.poll();
}
}
return res;
}
结尾
欢迎评论区交流,每日打卡,冲冲冲!!!