Leetcode 1792：最大平均通过率

题目描述

一所学校里有一些班级，每个班级里有一些学生，现在每个班都会进行一场期末考试。给你一个二维数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-average-pass-ratio
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    double maxAverageRatio(vector<vector<int>>& classes, int extraStudents) {
        auto cmp = [](pair<int, int> &a, pair<int, int> &b){
            long long p1 = a.first;
            long long t1 = a.second;
            long long p2 = b.first;
            long long t2 = b.second;
            return (t1*(p1+1) - p1*(t1+1))*t2*(t2+1) < (t2*(p2+1) - p2*(t2+1))*t1*(t1+1);
        };
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> que(cmp);
        for(auto &cla : classes) que.emplace(cla[0], cla[1]);
        for(int i = 1; i <= extraStudents; i++){
            auto [p, t] = que.top();
            que.pop();
            que.emplace(p+1, t+1);
        }
        double ans = 0.0;
        while(!que.empty()){
            auto [p, t] = que.top();
            que.pop();
            ans += static_cast<double>(p) / t;
        }
        return ans / classes.size();
    }
};