leetcode 5641.卡车上的最大单元数 Java

打卡博客链接

https://blog.youkuaiyun.com/qq_43349112/article/details/108542248

题目链接

https://leetcode-cn.com/problems/maximum-units-on-a-truck/

描述

请你将一些箱子装在 一辆卡车 上。给你一个二维数组 boxTypes ，其中 boxTypes[i] = [numberOfBoxesi, 
numberOfUnitsPerBoxi] ：

numberOfBoxesi 是类型 i 的箱子的数量。
numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
整数 truckSize 表示卡车上可以装载 箱子 的 最大数量 。只要箱子数量不超过 truckSize ，你就可以选择任意
箱子装到卡车上。

返回卡车可以装载 单元 的 最大 总数。


提示：

1 <= boxTypes.length <= 1000
1 <= numberOfBoxesi, numberOfUnitsPerBoxi <= 1000
1 <= truckSize <= 106

示例

示例 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

初始代码模板

class Solution {
    public int maximumUnits(int[][] boxTypes, int truckSize) {
    }
}

代码

由题可知，要尽可能取装载单元数大的箱子。
因此可以使用贪心思想，对箱子进行排序，装载单元数大的箱子先取。

并且因为有truckSize容量限制，因此不一定能取完，因此要进行判断。此时boxTypes已经排序
for i in boxTypes

• 如果当前剩余容量大于箱子数，那么都取，取之后答案加上boxTypes[i][0] * boxTypes[i][1]，容量减去箱子数
• 如果当前剩余容量小于小子书，那么只能去当前容量的箱子，取之后答案加上 剩余容量 * boxTypes[i][1]，退出循环

class Solution {
    public int maximumUnits(int[][] boxTypes, int truckSize) {
        //排序，为接下来的贪心做准备
        Arrays.sort(boxTypes, (o1, o2) -> o2[1] - o1[1]);

        int ans = 0;
        for (int i = 0; i < boxTypes.length; i++) {
            if (truckSize > boxTypes[i][0]) {
                ans += boxTypes[i][0] * boxTypes[i][1];
                truckSize -= boxTypes[i][0];
            } else {
                ans += truckSize * boxTypes[i][1];
                break;
            }
        }

        return ans;
    }
}