leetcode1792:最大平均通过率(1.19日每日一题)

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

写解题思路之前先来了解一下优先队列:

优先队列(priority_queue)比普通队列(queue)不同之处就是可以自定义其中数据的优先级,让优先级高的排在前面。

和队列基本操作一样:

top        访问队头元素

empty     判断队列是否为空

size        返回队列元素个数

push        插入元素到队尾(排序)

emplace  原地构造一个元素并插入队列

pop          弹出队头元素

swap        交换元素

priority_queue<int,vector<int>,greater<int>>q(从大到小排序,默认大顶堆)

priority_queue<int,vector<int>,less<int>> p(从小到大排序,小顶堆)

解题思路:

        我们需要把每个extraStudents找到最合适的班级来使平均通过率最大,每个extraStudents增加的通过率可以用加上的通过率然后再减去原来的通过率((classes[i][0]+1)/(double)(classes[i][1]+1)-(classes[i][0])/(double)(classes[i][1]))增大平均通过率的优先级就高了,将其队头元素的同学加+1,再次算出增长通过率,放到优先队列中。

        因为优先队列找到增长通过率最大的元素之和,还需要在原来的班级中加人数,将下标也放到优先队列之中。

解题代码:

class Solution {
public:
    double maxAverageRatio(vector<vector<int>>& classes, int extraStudents) {
        priority_queue<pair<double,int>>p;
        for(int i=0;i<classes.size();i++)
        {
            double values=(classes[i][0]+1)/(double)(classes[i][1]+1)-(classes[i][0])/(double)(classes[i][1]);
            p.emplace(values,i);
        }
        while(extraStudents)
        {
            auto[ave,i]=p.top();
            p.pop();
            classes[i][0]++;
            classes[i][1]++;
            double values=(classes[i][0]+1)/(double)(classes[i][1]+1)-(classes[i][0])/(double)(classes[i][1]);
            extraStudents--;
            p.emplace(values,i);
        }
        double studentPass=0;
        for(int i=0;i<classes.size();i++)
        {
            studentPass+=classes[i][0]/(double)(classes[i][1]);
        }
        return studentPass/classes.size();
    }
};

猜你喜欢

转载自blog.csdn.net/m0_63743577/article/details/129141630