雇佣 K 名工人的最低成本

857. 雇佣 K 名工人的最低成本 - 力扣（LeetCode）

有 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 <= 10^4
• 1 <= quality[i], wage[i] <= 10^4

解题

/**
 * @param {number[]} quality
 * @param {number[]} wage
 * @param {number} k
 * @return {number}
 */
var mincostToHireWorkers = function (quality, wage, k) {
  const len = quality.length;
  const idxs = new Array(len).fill(0).map((_, i) => i);
  idxs.sort((i, j) => wage[i] * quality[j] - wage[j] * quality[i]);
  const pq = [];
  let totalQ = 0;
  const findIndex = (q) => {
    let left = 0;
    let right = pq.length;
    while (left <= right) {
      const mid = (left + right) >> 1;
      if (pq[mid] === q) {
        return mid;
      }
      if (pq[mid] > q) {
        left = mid + 1;
      } else {
        right = mid - 1;
      }
    }
    return left;
  };
  for (let i = 0; i < k; i++) {
    const q = quality[idxs[i]];
    totalQ += q;
    let idx = findIndex(q);
    pq.splice(idx, 0, q);
  }
  let res = (totalQ * wage[idxs[k - 1]]) / quality[idxs[k - 1]];
  for (let i = k; i < len; i++) {
    const q = quality[idxs[i]];
    let idx = findIndex(q);
    if (idx > 0) {
      totalQ += q;
      pq.splice(idx, 0, q);
      totalQ -= pq.shift();
      res = Math.min(res, (totalQ * wage[idxs[i]]) / quality[idxs[i]]);
    }
  }
  return res;
};