1. 问题描述:
请你将一些箱子装在一辆卡车上。给你一个二维数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-units-on-a-truck
2. 思路分析:
首先需要理解题目表达的意思,我们需要求解的是箱子数目在truckSize的条件下可以装载单元的最大总数,很明显对于某一类型的箱子装载的单元数目越多那么肯定先装载这一类型的箱子,所以可以对boxTypes数组按照第二个元素从大到小进行排序,因为使用的是python语言所以可以使用lambda表达式规定排序的规则(python语言自定义排序的规则真的很方便),可以使用括号中参数的顺序来表示排序参数的顺序,某个参数添加负号表示对这个参数按照从大到小的规则进行排序
3. 代码如下:
from typing import List
class Solution:
def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
# python中可以使用lambda表达式可以自定义排序规则, 负号表示从大到小排序
boxTypes.sort(key=lambda x: (-x[1], x[0]))
# print(boxTypes)
cur, res = 0, 0
for i in range(len(boxTypes)):
if cur + boxTypes[i][0] < truckSize:
res += boxTypes[i][0] * boxTypes[i][1]
cur += boxTypes[i][0]
else:
res += (truckSize - cur) * boxTypes[i][1]
break
return res