【题目】
请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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 <= 106
【代码】
【Python】
思路:使用dict字典
执行用时:
48 ms, 在所有 Python3 提交中击败了92.08%的用户
内存消耗:
15.2 MB, 在所有 Python3 提交中击败了15.75%的用户
class Solution:
def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
rs=0
boxTypes=sorted(boxTypes, key = lambda kv:(kv[1], kv[0]),reverse=True)
for x in boxTypes:
if truckSize:
if x[0]<=truckSize:
truckSize-=x[0]
rs+=x[0]*x[1]
else:
rs+=truckSize*x[1]
truckSize=0
else:
break
return rs
【方法2】对内存友好,代码简洁
执行用时:
68 ms, 在所有 Python3 提交中击败了21.18%的用户
内存消耗:
14.9 MB, 在所有 Python3 提交中击败了94.01%的用户
class Solution:
def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
boxTypes=sorted(boxTypes, key = lambda kv:(kv[1], kv[0]),reverse=True)
rs=0
for x in boxTypes:
rs+=x[1]*min(x[0],truckSize)
truckSize-=min(x[0],truckSize)
return rs