1792. 最大平均通过率
难度中等69
一所学校里有一些班级,每个班级里有一些学生,现在每个班都会进行一场期末考试。给你一个二维数组 classes
,其中 classes[i] = [passi, totali]
,表示你提前知道了第 i
个班级总共有 totali
个学生,其中只有 passi
个学生可以通过考试。
给你一个整数 extraStudents
,表示额外有 extraStudents
个聪明的学生,他们 一定 能通过任何班级的期末考。你需要给这 extraStudents
个学生每人都安排一个班级,使得 所有 班级的 平均 通过率 最大 。
一个班级的 通过率 等于这个班级通过考试的学生人数除以这个班级的总人数。平均通过率 是所有班级的通过率之和除以班级数目。
请你返回在安排这 extraStudents
个学生去对应班级后的 最大 平均通过率。与标准答案误差范围在 10-5
以内的结果都会视为正确结果。
示例 1:
输入:classes = [[1,2],[3,5],[2,2]], extraStudents = 2
输出:0.78333
解释:你可以将额外的两个学生都安排到第一个班级,平均通过率为 (3/4 + 3/5 + 2/2) / 3 = 0.78333 。
示例 2:
输入:classes = [[2,4],[3,9],[4,5],[2,10]], extraStudents = 4
输出:0.53485
提示:
1 <= classes.length <= 105
classes[i].length == 2
1 <= passi <= totali <= 105
1 <= extraStudents <= 105
class Solution {
//https://leetcode.cn/problems/maximum-average-pass-ratio/solution/zui-da-ping-jun-tong-guo-lu-you-xian-dui-8mpm/
public double maxAverageRatio(int[][] classes, int extraStudents) {
//每次分配时应该分配给产生贡献最大的班级,从而使最终的贡献最大,是一种贪心策略。
int n = classes.length;
// 大根堆,堆顶是产生最大影响的班级
PriorityQueue<double[]> pq = new PriorityQueue<>((o1, o2) -> {
double x = ((o2[0] + 1) / (o2[1] + 1) - o2[0] / o2[1]);
double y = ((o1[0] + 1) / (o1[1] + 1) - o1[0] / o1[1]);
if (x > y) return 1;
if (x < y) return -1;
return 0;
});
for(int[] c : classes){
pq.offer(new double[]{c[0], c[1]});
}
// 每次分配一名学生
while(extraStudents > 0){
//取出能够产生最大影响的班级
double[] maxClass = pq.poll();
maxClass[0] += 1.0;
maxClass[1] += 1.0;
pq.offer(maxClass);
extraStudents--;
}
double res = 0;
while(!pq.isEmpty()){
double[] c = pq.poll();
res += c[0]/c[1];
}
return res/n;
}
}