【排序-简单】1710. 卡车上的最大单元数

最新推荐文章于 2022-11-15 11:39:28 发布

原创最新推荐文章于 2022-11-15 11:39:28 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

714 篇文章

订阅专栏

博客围绕将箱子装在卡车上的问题展开，给定二维数组boxTypes和卡车最大装载箱子数truckSize，求卡车可装载单元的最大总数。给出示例及提示，并展示了两种Python解法，分别说明了执行用时和内存消耗情况。

【题目】
请你将一些箱子装在一辆卡车上。给你一个二维数组 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
【代码】
【Python】
思路：使用dict字典
执行用时：
48 ms, 在所有 Python3 提交中击败了92.08%的用户
内存消耗：
15.2 MB, 在所有 Python3 提交中击败了15.75%的用户

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        rs=0        
        boxTypes=sorted(boxTypes, key = lambda kv:(kv[1], kv[0]),reverse=True)
        for x in boxTypes:
            if truckSize:
                if x[0]<=truckSize: 
                    truckSize-=x[0]
                    rs+=x[0]*x[1]
                else:
                    rs+=truckSize*x[1]
                    truckSize=0
            else:
                break
        return rs

【方法2】对内存友好，代码简洁
执行用时：
68 ms, 在所有 Python3 提交中击败了21.18%的用户
内存消耗：
14.9 MB, 在所有 Python3 提交中击败了94.01%的用户

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        boxTypes=sorted(boxTypes, key = lambda kv:(kv[1], kv[0]),reverse=True)
        rs=0
        for x in boxTypes:
            rs+=x[1]*min(x[0],truckSize)
            truckSize-=min(x[0],truckSize)
        return rs