You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.9 KiB
57 lines
1.9 KiB
package class_2023_02_4_week;
|
|
|
|
import java.util.PriorityQueue;
|
|
|
|
// 一所学校里有一些班级，每个班级里有一些学生，现在每个班都会进行一场期末考试
|
|
// 给你一个二维数组 classes ，其中 classes[i] = [passi, totali]
|
|
// 表示你提前知道了第 i 个班级总共有 totali 个学生，其中只有 passi 个学生可以通过考试
|
|
// 给你一个整数 extraStudents ，表示额外有 extraStudents 个聪明的学生
|
|
// 他们 一定 能通过任何班级的期末考
|
|
// 你需要给这 extraStudents 个学生每人都安排一个班级
|
|
// 使得 所有 班级的 平均 通过率 最大 。
|
|
// 一个班级的 通过率 等于这个班级通过考试的学生人数除以这个班级的总人数
|
|
// 平均通过率 是所有班级的通过率之和除以班级数目。
|
|
// 请你返回在安排这 extraStudents 个学生去对应班级后的 最大 平均通过率
|
|
// 与标准答案误差范围在 10^-5 以内的结果都会视为正确结果。
|
|
// 测试链接 : https://leetcode.cn/problems/maximum-average-pass-ratio/
|
|
public class Code01_MaximumAveragePassRatio {
|
|
|
|
public static double maxAverageRatio(int[][] classes, int extraStudents) {
|
|
// 堆 : 谁获得一个天才，得到的通过率增益最大
|
|
// 谁先弹出
|
|
PriorityQueue<Party> heap = new PriorityQueue<>((a, b) -> a.benefit() - b.benefit() < 0 ? 1 : -1);
|
|
for (int[] p : classes) {
|
|
heap.add(new Party(p[0], p[1]));
|
|
}
|
|
Party cur;
|
|
// 一个一个天才分配
|
|
for (int i = 0; i < extraStudents; i++) {
|
|
cur = heap.poll();
|
|
cur.pass++;
|
|
cur.total++;
|
|
heap.add(cur);
|
|
}
|
|
double all = 0;
|
|
while (!heap.isEmpty()) {
|
|
cur = heap.poll();
|
|
all += cur.pass / cur.total;
|
|
}
|
|
return all / classes.length;
|
|
}
|
|
|
|
public static class Party {
|
|
public double pass;
|
|
public double total;
|
|
|
|
public Party(int p, int t) {
|
|
pass = p;
|
|
total = t;
|
|
}
|
|
|
|
public double benefit() {
|
|
return (pass + 1) / (total + 1) - pass / total;
|
|
}
|
|
}
|
|
|
|
}
|