java 优先队列 leetcode应用

最新推荐文章于 2024-09-08 20:02:08 发布

刘柯洋

最新推荐文章于 2024-09-08 20:02:08 发布

阅读量68

点赞数

文章标签： leetcode 算法 java Powered by 金山文档

本文链接：https://blog.csdn.net/m0_52049037/article/details/129118892

版权

文章描述了一道LeetCode的编程问题，目标是通过合理分配额外的聪明学生到各个班级，来最大化所有班级的平均通过率。关键策略是使用优先队列，根据通过率的增益来决定学生的分配顺序。每次分配时，计算并更新班级的通过率，然后重新插入优先队列中。

摘要由CSDN通过智能技术生成

leetcode1792. 最大平均通过率

一所学校里有一些班级，每个班级里有一些学生，现在每个班都会进行一场期末考试。给你一个二维数组 classes ，其中 classes[i] = [passi, totali] ，表示你提前知道了第 i 个班级总共有 totali 个学生，其中只有 passi 个学生可以通过考试。

给你一个整数 extraStudents ，表示额外有 extraStudents 个聪明的学生，他们一定能通过任何班级的期末考。你需要给这 extraStudents 个学生每人都安排一个班级，使得所有班级的平均通过率最大。

一个班级的通过率等于这个班级通过考试的学生人数除以这个班级的总人数。平均通过率是所有班级的通过率之和除以班级数目。

请你返回在安排这 extraStudents 个学生去对应班级后的最大平均通过率。与标准答案误差范围在 10-5 以内的结果都会视为正确结果。

来源：力扣（LeetCode）

链接：https://leetcode.cn/problems/maximum-average-pass-ratio

思想

由于班级总数不会变化，因此题目所求「最大化平均通过率」等价于「最大化总通过率」。设某个班级的人数为 total，其中通过考试的人数为 pass，那么给这个班级安排一个额外通过考试的学生，其通过率会增加：

$\text{[math]}$

我们会优先选择通过率增加量最大的班级，这样做的正确性在于给同一个班级不断地增加安排的学生数量时，其增加的通过率是单调递减的，即：

$\text{[math]}$

因此当以下条件满足时，班级 j 比班级 i 优先级更大：

$\text{[math]}$

化简后可得：

$\text{[math]}$

我们按照上述比较规则将每个班级放入优先队列中，进行 extraStudents 次操作。每一次操作，我们取出优先队列的堆顶元素，令其 pass 和 total 分别加 1，然后再放回优先队列。

题解中优先队列的初始化

PriorityQueue<int[]> pq = new PriorityQueue<int[]>((a, b) -> {
            long val1 = (long) (b[1] + 1) * b[1] * (a[1] - a[0]);
            long val2 = (long) (a[1] + 1) * a[1] * (b[1] - b[0]);
            if (val1 == val2) {
                return 0;
            }
            return val1 < val2 ? 1 : -1;
        });