刷题的日常-卡车上的最大单元数

84 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 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;
    }
}

image.png