目标
一所学校里有一些班级,每个班级里有一些学生,现在每个班都会进行一场期末考试。给你一个二维数组 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 <= 10^5
- classes[i].length == 2
- 1 <= passi <= totali <= 10^5
- 1 <= extraStudents <= 10^5
思路
有一个二维数组 classes
, classes[i][0]
表示班级 i
期末考试中通过考试的人数,classes[i][1]
表示班级 i
的总人数。有 extraStudents
个聪明学生一定可以通过期末考试,现在需要将这些学生分配到班级中去,使得班级通过率的平均值最大。返回最大平均通过率。
为了使平均值更大,可以优先将聪明学生安排到通过率提升最大的班级,使用优先队列。
代码
/**
* @date 2025-09-01 21:21
*/
public class MaxAverageRatio1792 {
public double maxAverageRatio(int[][] classes, int extraStudents) {
PriorityQueue<int[]> q = new PriorityQueue<>((a, b) -> (int) (((b[1] - b[0]) * (long) a[1] * (a[1] + 1) - (a[1] - a[0]) * (long) b[1] * (b[1] + 1)) % 1000000007));
for (int[] item : classes) {
q.offer(item);
}
for (int i = 0; i < extraStudents; i++) {
int[] peek = q.poll();
peek[0]++;
peek[1]++;
q.offer(peek);
}
double res = 0.0;
for (int[] item : classes) {
res += (double) item[0] / item[1];
}
return res / classes.length;
}
}