请你将一些箱子装在 一辆卡车 上。给你一个二维数组 boxTypes ,其中 boxTypes[i] = [numberOfBoxesi, numberOfUnitsPerBoxi] :
- numberOfBoxesi 是类型 i 的箱子的数量。
- numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
整数 truckSize 表示卡车上可以装载 箱子 的 最大数量 。只要箱子数量不超过 truckSize ,你就可以选择任意箱子装到卡车上。
返回卡车可以装载 单元 的 最大 总数。
示例 1:
输入:boxTypes = [[1,3],[2,2],[3,1]], truckSize = 4
输出:8
解释:箱子的情况如下:
- 1 个第一类的箱子,里面含 3 个单元。
- 2 个第二类的箱子,每个里面含 2 个单元。
- 3 个第三类的箱子,每个里面含 1 个单元。 可以选择第一类和第二类的所有箱子,以及第三类的一个箱子。 单元总数 = (1 * 3) + (2 * 2) + (1 * 1) = 8
示例 2:
输入:boxTypes = [[5,10],[2,5],[4,7],[3,9]], truckSize = 10
输出:91
提示:
- 1 <= boxTypes.length <= 1000
- 1 <= numberOfBoxesi, numberOfUnitsPerBoxi <= 1000
- 1 <= truckSize <= 10^6
我的想法:
1.按着题意,它需要最大的总数,那么就需要将 numberOfUnitsPerBoxi (是类型 i 每个箱子可以装载的单元数量)从大到小排序。
2.新搞一个 returnsum = 0,循环排完序的列表,比较 truckSize 和 numberOfBoxesi 的大小:
如果 truckSize 大于等于 numberOfBoxesi 的话,returnsum += numberOfBoxesi * numberOfUnitsPerBoxi,然后用 truckSize -= numberOfBoxesi ;
如果 truckSize 小于 numberOfBoxesi 的话,returnsum += truckSize * numberOfUnitsPerBoxi,退出循环。
class Solution:
def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
boxTypes.sort(key = lambda x : (x[1],x[0]),reverse = True)
returnsum = 0
for box in boxTypes:
if truckSize >= box[0]:
returnsum += box[0] * box[1]
truckSize -= box[0]
else:
returnsum += truckSize * box[1]
break
return returnsum
粘几个题解:
官方的思路也差不多 - 贪心
【宫水三叶】简单贪心