【C++堆(优先队列)】1792. 最大平均通过率|1817 不知道如何证明，欢迎告诉我-CSDN博客

本文链接：https://blog.csdn.net/he_zhidan/article/details/142317384

本文涉及知识点

LeetCode1792. 最大平均通过率

一所学校里有一些班级，每个班级里有一些学生，现在每个班都会进行一场期末考试。给你一个二维数组 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⁵
classes[i].length == 2
1 <= passi <= totali <= 10⁵
1 <= extraStudents <= 10⁵

堆（优先队列）

由于班级数不变，所以平均通过率最大 $\iff$ 总通过率最大。
某班有b个同学，a人通过，则工作率增加d = (a+1.0) $\div$ (b+1)-(double)a/b
将{d,a,b}放到堆中，将extraStudents同学依次放到堆顶元素：
即a++,b++，并更新d。
注意：堆顶元素不能修改，只能弹出后，加入。
证明过程不会，求教中。

代码

核心代码

class Solution {
		public:
			double maxAverageRatio(vector<vector<int>>& classes, int extraStudents) {
				priority_queue<tuple<double, int, int>> heap;
				auto Add = [&](int a, int b) {
					return (a + 1.0) / (b + 1) - (double)a / b;
				};
				for (const auto& v : classes) {
					heap.emplace(Add(v[0], v[1]), v[0], v[1]);
				}
				while (extraStudents--) {
					auto [tmp, a, b] = heap.top();
					heap.pop();
					a++, b++;
					heap.emplace(Add(a, b), a, b);
				}
				double ret = 0;
				while (heap.size()) {
					auto [tmp, a, b] = heap.top();
					heap.pop();
					ret += (double)a / b;
				}
				return ret / classes.size();
			}
		};

单元测试

vector<vector<int>> classes;
		int extraStudents;
		TEST_METHOD(TestMethod11)
		{
			classes = { {1,2},{3,5},{2,2} }, extraStudents = 2;
			auto res = Solution().maxAverageRatio(classes, extraStudents);
			AssertEx(0.78333, res);
		}
		TEST_METHOD(TestMethod12)
		{
			classes = { {2,4},{3,9},{4,5},{2,10} }, extraStudents = 4;
			auto res = Solution().maxAverageRatio(classes, extraStudents);
			AssertEx(0.53485, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功