【Leetcode】857. Minimum Cost to Hire K Workers

题目地址：

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

给定$n$个工人，其有两个属性，一个是质量，一个是期望工资，以长$n$数组$q$和$w$表示。给定一个正整数$k,k\le n$，要求雇佣恰好$k$名工人，使得这$k$个工人，每个工人的薪资与其$q$值成比例，并且每个工人的薪资不低于其期望薪资。问满足条件的情况下，最少总共要付多少薪资。

由于要按照$q$值的比例分配工资，设比例是$t$，则挑出的$k$个人的工资分别是$q_{1}t,q_{2}t,...,q_{k}t$，同时需要满足$\forall i,q_{i}t\ge w_i$，又要使得总薪资最小，所以$t$应该要尽量小，所以 t = max ⁡ 1 ≤ i ≤ k { w i / q i } t=\max_{1\le i\le k}\{w_i/q_i\} t=max1≤i≤k​{wi​/qi​}，即$t$必然等于某一个$w_i/q_i$。可以先将所有的工人按照$w_i/q_i$从小到大排序，然后从小到大枚举$t$可以取的值，即枚举$t$取$t_i=w_i/q_i$的情形，由于总工资等于$t\sum q_i$，所以要在$t$值小于等于$t_i$的工人里找最小的$k$个$q$值，这可以用最大堆来做，该最大堆维护的是$q$值最小的$k$个人，同时需要一个变量维护堆内的$q$值和，方便算总工资。由于我们是从小到大枚举$t_i$的，所以堆里存的都是$t$值小于$t_i$的工人，此外，由最大堆性质，我们还可以维持堆里存的是$q$值最小的$k$个工人。代码如下：

import java.util.ArrayList;
import java.util.List;
import java.util.PriorityQueue;

public class Solution {
    
    class Pair {
        double ratio;
        int quality;
        
        public Pair(double ratio, int quality) {
            this.ratio = ratio;
            this.quality = quality;
        }
    }
    
    public double mincostToHireWorkers(int[] quality, int[] wage, int k) {
        int n = quality.length;
        List<Pair> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            list.add(new Pair((double) wage[i] / quality[i], quality[i]));
        }
        // 按照t值从小到大排序
        list.sort((p1, p2) -> Double.compare(p1.ratio, p2.ratio));
        // 堆里存所以t值小于等于当前枚举的t值的工人中，q值最小的k个人的q值
        PriorityQueue<Integer> minHeap = new PriorityQueue<>((x, y) -> -Integer.compare(x, y));
        double res = 1e18, sum = 0;
        for (Pair pair : list) {
            sum += pair.quality;
            minHeap.offer(pair.quality);
            if (minHeap.size() > k) {
                sum -= minHeap.poll();
            }
            
            if (minHeap.size() == k) {
                res = Math.min(res, sum * pair.ratio);
            }
        }
        
        return res;
    }
}

时间复杂度$O(n\log n)$，空间$O(n)$。