题目描述:
请你将一些箱子装在 一辆卡车 上。给你一个二维数组 boxTypes ,其中 boxTypes[i] = [numberOfBoxesi, numberOfUnitsPerBoxi] :
numberOfBoxesi 是类型 i 的箱子的数量。
numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
整数 truckSize 表示卡车上可以装载 箱子 的 最大数量 。只要箱子数量不超过 truckSize ,你就可以选择任意箱子装到卡车上。
返回卡车可以装载 单元 的 最大 总数。
解题思路:
根据题意,我们应该选择尽可能多的单元数,因此,我们先对 boxTypes 按照单元数从大到小的顺序排列。
然后从前往后遍历 boxTypes,选择最多 truckSize 个箱子,累加单元数。
时间复杂度 O(n×logn),其中 n 表示二维数组 boxTypes 的长度。
代码演示:
class Solution:
def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
ans = 0
for a, b in sorted(boxTypes, key=lambda x: -x[1]):
ans += b * min(truckSize, a)
truckSize -= a
if truckSize <= 0:
break
return ans
学习产出:
对贪心算法有了更深入的理解