LeetCode 1792 最大平均通过率结构体 堆HERODING的LeetCode之路

Posted 2023-03-04 HERODING23

解题思路：
选择这道题并不是因为这道题有多难，其实只要看完题目，解题思路基本上都能浮现在脑海中，用堆去存储通过率，排序标准是通过率经过+1变换能够有最大幅度的增加，这一点并不好实现，通用的数据结构没法实现这个功能。那么就需要我们自行定义一个数据结构，该数据结构中存储pass和total两个变量，并且对符号<进行一个重载，判断逻辑修改成最大幅度的增加将会排序在前面。步骤如下：

1. 定义结构体Ratio；
2. 将classes入堆；
3. 遍历extrastudents次，将堆顶的ratio进行更新；
4. 累加所有班级的通过率，返回平均结果。

代码如下：

class Solution 
public:
    // 自定义结构体
    struct Ratio 
        int pass, total;
        // 选择通过率增加最大的
        bool operator < (const Ratio& r) const 
            return (long long) (r.total + 1) * r.total * (total - pass) < (long long) (total + 1) * total * (r.total - r.pass);
        
    ;

    double maxAverageRatio(vector<vector<int>>& classes, int extraStudents) 
        priority_queue<Ratio> q;
        for(auto& c : classes) 
            q.push(c[0], c[1]);
        
        for(int i = 0; i < extraStudents; i ++) 
            auto [pass, total] = q.top();
            q.pop();
            q.push(pass + 1, total + 1);
        
        double res = 0.0;
        while(!q.empty()) 
            auto [pass, total] = q.top();
            q.pop();
            res += 1.0 * pass / total;
        
        return res / classes.size();
    
;