Leetcode_1710. 卡车上的最大单元数

【原题链接】

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

示例 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(vector<vector<int>>& boxTypes, int truckSize) {
        map<int, int, greater<int>> mp;
        for(auto x : boxTypes) mp[x[1]] += x[0];//x[1]的大小排序
        int res = 0;
        for(auto x : mp){
            int tmp = min(x.second, truckSize);
            truckSize -= tmp;
            res += tmp * x.first;
        }
        return res;
    }
};

执行用时: 100 ms

内存消耗: 18.9 MB

【时间复杂度】

O(nlogn)

【注意点】

  1. 其他解法:对vector直接排序

  2. 对vector直接排序很慢,操作如下:

sort(boxType.begin(), boxType.end(), [](vector<int>a, vector<int> b){
 return a[1] > b[1];
})

评论