题目链接:https://leetcode.cn/problems/maximum-units-on-a-truck/Q
题目描述:
请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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
解法:排序 + 贪心
先按照 每个箱子可以装载的单元数量将箱子类型进行降序排序,再利用排好序后的数组,依次取出箱子将其可装载的单位数量加到结果中。
代码:
class Solution {
public int maximumUnits(int[][] boxTypes, int truckSize) {
int result = 0;
Arrays.sort(boxTypes, (a, b) -> b[1] - a[1]);
for (int i = 0; i < boxTypes.length; i++) {
if (truckSize > 0) {
result += (truckSize - boxTypes[i][0] >= 0 ? boxTypes[i][0] : truckSize) * boxTypes[i][1];
truckSize = truckSize - boxTypes[i][0] >= 0 ? truckSize - boxTypes[i][0] : 0;
} else {
break;
}
}
return result;
}
}
网友评论