1792 - 最大平均通过率 - 贪心 - 数学 - 堆

欢迎关注更多精彩
关注我，学习常用算法与数据结构，一题多解，降维打击。

题目描述

[1792] 最大平均通过率

https://leetcode-cn.com/problems/maximum-average-pass-ratio/

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

Related Topics

• 贪心
• 数学
• 堆

题目剖析&信息挖掘

此题为规律题，需要通过数学分析得到规则。

解题思路

方法一 贪心+堆+数学

解析

以例1为例。

classes = [[1,2],[3,5],[2,2]], extraStudents = 2

假设现在extraStudents=1

我们有3种选择，放1班，放2班，放3班。

结果如下：

由于总班级（分母）不变，我们只计算总平均分

放1班：(1+1)/（2+1） + 3/5 + 2/2 = 2.266667

放2班：1/2 + (3+1)/(5+1) + 2/2 = 2.166667

放3班：1/2 + 3/5+ （2+1）/（2+1）= 2.1

说明放在1班最好。

同理再来extraStudents=2，在extraStudents=1的基础上再继续选择。

复杂度分析，每次选择都要O(n), 一共做N次，复杂度为O(n^2)。

优化

从新分析上述过程

放1班：(1+1)/（2+1） + 3/5 + 2/2 = 1/2 + 3/5 + 2/2 + ((1+1)/（2+1) - 1/2) = 2.266667

放2班：1/2 + (3+1)/(5+1) + 2/2 = 1/2 + 3/5 + 2/2 + ( (3+1)/(5+1) - 3/5 ) = 2.166667

放3班：1/2 + 3/5+ （2+1）/（2+1）= 1/2 + 3/5 + 2/2 + (（2+1）/（2+1）- 2/2) = 2.1

可以看出在未选择前大家三种情况分数是一样的，选择放到一个班级后，其实就是在原来的分数上加上一个差值。

放第i班增加的分数(边际效益) diff[i] = (class[i][0]+1)/ (class[i][1]+1) - class[i][0]/class[i][1]

每次都选择边际效益最大的那个班级加1.

可以使用大顶堆查询到最大的边际效益的班级。

思路

func maxAverageRatio(classes [][]int, extraStudents int) float64 {
	h := &myHeap{} // 大顶堆

	for _, c := range classes { // 将现有班级数据加入到堆中
		heap.Push(h, [2]float64{float64(c[0]), float64(c[1])})
	}

	for ; extraStudents > 0; extraStudents-- {
		t :=heap.Pop(h).([2]float64)  // 每次拿出堆顶数据，此班级边际效益最大
		// 将些班同学+1
		t[0]++
		t[1]++
		heap.Push(h, t) // 再放回堆中
	}

	// 统计所有班级总平均分
	sum :=float64(0)
	for h.Len()>0 {
		t :=heap.Pop(h).([2]float64)
		sum += t[0]/t[1]
	}

	return sum/float64(len(classes)) // 需要除以总班级
}

注意

• 题目要计算的是各班级的平均分的总平均分。

知识点

• 贪心
• 数学
• 堆

复杂度

• 时间复杂度：O(nlog(n))
• 空间复杂度：O(n)

代码实现

type myHeap [][2]float64

func (h *myHeap) Less(i, j int) bool {
	s1 := ((*h)[i][0]+1) / ((*h)[i][1]+1) - (*h)[i][0] / (*h)[i][1] //计算边际效益
	s2 := ((*h)[j][0]+1) / ((*h)[j][1]+1) - (*h)[j][0] / (*h)[j][1]
	return s1 > s2 // 边际效益大的到顶上
}

func (h *myHeap) Swap(i, j int) {
	(*h)[i], (*h)[j] = (*h)[j], (*h)[i]
}

func (h *myHeap) Len() int {
	return len(*h)
}

func (h *myHeap) Pop() (v interface{}) {
	*h, v = (*h)[:h.Len()-1], (*h)[h.Len()-1]
	return
}

func (h *myHeap) Push(v interface{}) {
	*h = append(*h, v.([2]float64))
}

func maxAverageRatio(classes [][]int, extraStudents int) float64 {
	h := &myHeap{} // 大顶堆

	for _, c := range classes { // 将现有班级数据加入到堆中
		heap.Push(h, [2]float64{float64(c[0]), float64(c[1])})
	}

	for ; extraStudents > 0; extraStudents-- {
		t :=heap.Pop(h).([2]float64)  // 每次拿出堆顶数据，此班级边际效益最大
		// 将些班同学+1
		t[0]++
		t[1]++
		heap.Push(h, t) // 再放回堆中
	}

	// 统计所有班级总平均分
	sum :=float64(0)
	for h.Len()>0 {
		t :=heap.Pop(h).([2]float64)
		sum += t[0]/t[1]
	}

	return sum/float64(len(classes)) // 需要除以总班级
}

---

本人码农，希望通过自己的分享，让大家更容易学懂计算机知识。