题目来源

1710. 卡车上的最大单元数（LeetCode）

题目描述（简单）

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

题目解析

方法一：贪心

只能装 truckSize 个箱子到卡车上，依据贪心的思路，只需要每次都拿到剩余数组中单元数最大的箱子即可。boxTypes 只需按照 numberOfUnitsPerBoxi 进行逆序排序，在进行遍历累加即可。

代码

/**
 * @param {number[][]} boxTypes
 * @param {number} truckSize
 * @return {number}
 */
var maximumUnits = function(boxTypes, truckSize) {
    boxTypes.sort(function(left,right){return right[1]-left[1]})
    let sum = 0
    boxTypes.forEach((n)=>{
        if(n[0]<=truckSize){
            truckSize-=n[0]
            sum+=n[0]*n[1]
        }else if(truckSize!==0){
            sum+=truckSize*n[1]
            truckSize=0
        }
    })
    return sum
};

如图

官方代码

var maximumUnits = function(boxTypes, truckSize) {
    boxTypes.sort((a, b) => b[1] - a[1]);
    let res = 0;
    for (const boxType of boxTypes) {
        let numberOfBoxes = boxType[0];
        let numberOfUnitsPerBox = boxType[1];
        if (numberOfBoxes < truckSize) {
            res += numberOfBoxes * numberOfUnitsPerBox;
            truckSize -= numberOfBoxes;
        } else {
            res += truckSize * numberOfUnitsPerBox;
            break;
        }
    }
    return res;
};

如图

优化

由于forEach不能跳出循环，所以采用for...fo...对代码进行优化。

/**
 * @param {number[][]} boxTypes
 * @param {number} truckSize
 * @return {number}
 */
var maximumUnits = function(boxTypes, truckSize) {
    boxTypes.sort(function(left,right){return right[1]-left[1]})
    let sum = 0
   for(const n of boxTypes){
        if(n[0]<=truckSize){
            truckSize-=n[0]
            sum+=n[0]*n[1]
        }else{
            sum+=truckSize*n[1]
            break
        }
    }
    return sum
};

如图

参考

• 卡车上的最大单元数 - 力扣（LeetCode）

执行用时和内存消耗仅供参考，大家可以多提交几次。如有更好的想法，欢迎大家提出。