public double mincostToHireWorkers(int[] qs, int[] ws, int k) {
//由题意知,对于每个人,总金额>=总质量*单位质量工资
int n = qs.length;
double[][] ds = new double[n][2];//存储单位质量工资和质量
for (int i = 0; i < n; i++) {
ds[i][0] = ws[i] * 1.0 / qs[i]; ds[i][1] = qs[i];
}
Arrays.sort(ds, (a,b)->Double.compare(a[0], b[0]));//按单位质量工资从小到大排序
PriorityQueue<Integer> q = new PriorityQueue<>((a,b)->b-a);//质量大根堆
double ans = 1e18;
//枚举最大的单位质量工资,寻找k人的最小总质量
for (int i = 0, tot = 0; i < n; i++) {
int cur = (int)ds[i][1];
tot += cur; q.add(cur);//更新总质量
if (q.size() > k) tot -= q.poll();//超过k人取出最大质量
if (q.size() == k) ans = Math.min(ans, tot * ds[i][0]);
}
return ans;
}
2022.9.11-----leetcode.857
最新推荐文章于 2024-08-26 23:00:00 发布