1710. 卡车上的最大单元数

最新推荐文章于 2024-08-10 22:52:22 发布

月洇.

最新推荐文章于 2024-08-10 22:52:22 发布

阅读量98

点赞数 1

分类专栏：力扣文章标签： leetcode python

本文链接：https://blog.csdn.net/y0987654321yr/article/details/127872987

版权

力扣专栏收录该内容

23 篇文章 0 订阅

订阅专栏

请你将一些箱子装在一辆卡车上。给你一个二维数组 boxTypes ，其中 boxTypes[i] = [numberOfBoxesi, numberOfUnitsPerBoxi] ：

numberOfBoxesi 是类型 i 的箱子的数量。
numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
整数 truckSize 表示卡车上可以装载箱子的最大数量。只要箱子数量不超过 truckSize ，你就可以选择任意箱子装到卡车上。

返回卡车可以装载单元的最大总数。

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        box = sorted(boxTypes, key=lambda x: -x[1])
        res = 0
        for item in box:
            truckSize -= item[0]
            if truckSize > 0:
                res += item[0] * item[1]
            else:
                res += item[1] * (truckSize + item[0])
                return res
        return res