开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第6天,点击查看活动详情
刷题的日常-2022年11月26号
一天一题,保持脑子清爽
卡车上的最大单元数
来自leetcode的 1710 题,题意如下:
请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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
理解题意
通过题意,我们可以将信息整理如下:
- 题目给出一个二维数组
- 数组里的每个元素包含两个值,第一个代表箱子的数量,第二个代表箱子的包含的单元
- 再给定一个箱子数,要求卡车能装载的最大单元数量
做题思路
简单题,其实用贪心就能解。给定卡车能够装进去的箱子,要求单元数最大,那么每次获取的时候就应该拿单元数最多的箱子即可。步骤如下:
- 将二维数组以单元数量大小进行排序
- 从后往前遍历,如果当前剩余的箱子数量还能放的下,则把单元数乘以可以放进去的箱子数量
- 加上当前放进去的单元数
- 返回放进去的结果即可
代码实现
代码实现如下:
public class Solution {
public int maximumUnits(int[][] boxTypes, int truckSize) {
Arrays.sort(boxTypes, Comparator.comparingInt(o -> o[1]));
int result = 0;
for (int i = boxTypes.length - 1; i >= 0 && truckSize > 0; i--) {
result += Math.min(boxTypes[i][0], truckSize) * boxTypes[i][1];
truckSize -= boxTypes[i][0];
}
return result;
}
}