LC1710-卡车上的最大单元数

题目名称：卡车上的最大单元数

请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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 <= 10^6$

思路分析

对数组boxTypes按照第二列降序排序，这样之后从i=0开始遍历时就能保证先拿装载数量单元多的箱子，这样收益才能最大(这里我还考虑了第二列如果一样的情况，但其实不考虑也行)；从头开始遍历数组boxTypes，boxTypes[i][0]为第i类箱子的数量：如果此时卡车上还可以装载箱子的数量truckSize大于(等于)boxTypes[i][0]，则收益为boxTypes[i][0]*boxTypes[i][1]，同时需要更新truckSize;

如果此时卡车上还可以装载箱子的数量truckSize已经小于boxTypes[i][0]，则收益为truckSize*boxTypes[i][1]，此时可直接返回res,因卡车上已经不能再装载箱子。

Code实现

public int maximumUnits(int[][] boxTypes, int truckSize) {
    int res = 0;
    /**
     * 第二列是单元数 按照第二列降序排列
     * 如果第二列一样 按第一列降序排列
     */
    Arrays.sort(boxTypes, (e1, e2) - > (e1[1] == e2[1] ? (e2[0] - e1[0]) : (e2[1] - e1[1])));
    for (int i = 0; i < boxTypes.length; i++) {
        /**
         *  boxTypes[i][0]为箱子的数量
         *  boxTypes[i][1]为一个箱子的单元数
         */
        int cnt = boxTypes[i][0];
        if (truckSize >= cnt) {
            res += cnt * boxTypes[i][1];
            truckSize -= cnt;
        } else {
            res += truckSize * boxTypes[i][1];
            return res;
        }
    }

    return res;
}
复制代码

结果

算法复杂度分析

时间复杂度： $O(n)$
空间复杂度： $O(1)$