1710. Maximum Units on a Truck (Easy)

更新于 阅读次数： 本文字数： 389 阅读时长 ≈ 1 分钟

卡车装箱问题。你有一辆卡车和一个货物清单（2D 数组），货物清单每个项目列出了这种货物的数量和每个货物所占的容量。

你需要求出给定尺寸的卡车最多能装下多少个货物。这道题运用贪心算法。

思路

将 boxTypes 按照 units 排序，从最大的 units 开始装入卡车，直到装满 truckSize 的数量。

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        boxTypes.sort(key=lambda x: x[1], reverse=True)
        ans = 0
        for nums, units in boxTypes:
            if truckSize <= nums:
                return ans + truckSize * units
            else:
                ans += nums * units
                truckSize -= nums
        return ans
相关文章