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

题目链接：
https://leetcode-cn.com/problems/maximum-units-on-a-truck

解题思路：
1.按照盒子数量排序
2.从头开始遍历，计算可以装的盒子数量和单元数量

代码：

class Solution:
    def maximumUnits(self, boxTypes, truckSize):
        boxTypes.sort(key=lambda x:x[1],reverse=True)
        box_num = boxTypes[0][0]
        result = boxTypes[0][0] * boxTypes[0][1]
        if box_num > truckSize:
            return (truckSize * boxTypes[0][1])
        for i in range(1,len(boxTypes)):
            if box_num + boxTypes[i][0] < truckSize:
                box_num += boxTypes[i][0]
                result += boxTypes[i][0] * boxTypes[i][1]
            else:
                result += boxTypes[i][1] * (truckSize - box_num)
                break
        return result
      
s = Solution()
print(s.maximumUnits([[5,10],[2,5],[4,7],[3,9]],10))

写法二（更简洁）

class Solution:
    def maximumUnits(self, boxTypes, truckSize):
        boxTypes.sort(key=lambda x:x[1],reverse=True)
        res = 0
        for num,size in boxTypes:
            cnt = min(truckSize,num) #当前可以转的箱子数量
            res += cnt * size
            truckSize -= cnt #剩余可以装的箱子数量
            if truckSize == 0:
                break
        return res
      
s = Solution()
print(s.maximumUnits([[5,10],[2,5],[4,7],[3,9]],10))