题型为多路归并
我们的关注点应该是每个班级增加一个通过的人的时候,其通过率的变化,也就是增量。
假设当前班级b个通过的人,总共a个人,假设为此班级增加1个人,那么其通过率的增量应该为
b
+
1
a
+
1
−
b
a
=
a
−
b
a
(
a
+
1
)
\frac{b+ 1}{a+ 1}-\frac{b}{a}=\frac{a-b}{a(a+1)}
a+1b+1−ab=a(a+1)a−b,通过这个式子我们可以发现,每次增加一个人,其通过率肯定是增加的,但是每次增加的量,相比较上一次增加一人是减少的(分子a-b是个常数,而随着增加的人数变多,a逐渐增大,所以增量呈递减趋势)。
针对这一规律,我们可以先列出每个班级通过人数的增量的变化所对应的序列:
假设第i个班级通过的人数和总人数分别为 b i 与 a i b_i 与a_i bi与ai
第几个班级 | 增量序列(分别增加1个人,2个人,3个人…等对应的序列) |
---|---|
1 | a 1 − b 1 a 1 ( a 1 + 1 ) \frac{a_1-b_1}{a_1(a_1+1)} a1(a1+1)a1−b1 , a 1 − b 1 ( a 1 + 1 ) ( a 1 + 2 ) \frac{a_1-b_1}{(a_1+1)(a_1+2)} (a1+1)(a1+2)a1−b1, a 1 − b 1 ( a 1 + 2 ) ( a 1 + 3 ) \frac{a_1-b_1}{(a_1+2)(a_1+3)} (a1+2)(a1+3)a1−b1 … |
2 | a 2 − b 2 a 2 ( a 2 + 1 ) \frac{a_2-b_2}{a_2(a_2+1)} a2(a2+1)a2−b2 , a 2 − b 2 ( a 2 + 1 ) ( a 2 + 2 ) \frac{a_2-b_2}{(a_2+1)(a_2+2)} (a2+1)(a2+2)a2−b2, a 2 − b 2 ( a 2 + 2 ) ( a 2 + 3 ) \frac{a_2-b_2}{(a_2+2)(a_2+3)} (a2+2)(a2+3)a2−b2 … |
如上我们可以写出每个班级的增量的序列,并且这个序列是递减趋势的。
下文用x代替extraStudent
而总共可以有x个学生可以让我们添加到某些班级去,也就意味着我们只能从上述序列中选出x项作为通过率的增量,那么到这里思路就变得简单了:
我们要做的就是选择这些序列中前x个最大的增量,这一点可以通过优先级队列实现(排序的根据是每个班级增加一个人时通过率对应的增量)。每次找到增量最大的班级并取出来,更新后将其在放回队列,取x次即可。
class Solution {
public double maxAverageRatio(int[][] classes, int extraStudents) {
double res=0;
PriorityQueue<Cla>q=new PriorityQueue<>();
for(int []t:classes)
{
int a=t[1];
int b=t[0];
//注意,这里要写成1.0,不然会有精度的缺失
double d=(double)(a-b)/(a*(a+1.0));
res+=(double)b/a;
q.add(new Cla(a,b,d));
}
while(extraStudents!=0)
{
extraStudents--;
Cla cur= q.poll();
cur.a+=1;
cur.b+=1;
int a=cur.a;
int b=cur.b;
//说明这个班级要增加d这么多通过率
res+=cur.d;
cur.d=(double)(a-b)/(double)(a*(a+1.0));
q.add(cur);
}
return res/classes.length;
}
static class Cla implements Comparable<Cla>
{
//增量
double d;
//总人数与通过人数
int a,b;
public Cla(int a,int b,double d)
{
this.a=a;this.b=b;
this.d=d;
}
@Override
public int compareTo(Cla o1) {
// TODO Auto-generated method stub
if(this.d<o1.d)
{
return 1;
}
return -1;
}
}
}