LeetCode 5641. 卡车上的最大单元数 做题小结

题目

请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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) {
        Arrays.sort(boxTypes,(a,b) -> a[1]>b[1]?-1:1);
        int num;
        int ans = 0;
        for(int i=0;i<boxTypes.length&&truckSize>0;i++){
            num=boxTypes[i][0];
            num = Math.min(truckSize,num);
            ans+=num * boxTypes[i][1];
            truckSize-=num;
        }
        return ans;
    }
}