贪心算法例题

# 请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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
def zhuangxiangzi(boxTypes,truckSize):
    boxTypes.sort(key=lambda x:x[1], reverse=True)
    print(boxTypes)
    m=[0 for _ in range(len(boxTypes))]
    print(m)
    for i,[a, b] in enumerate(boxTypes):#a是个数，B是容量
        if truckSize>a:
            m[i]=b*a
            truckSize=truckSize-a
        else:
            m[i]=b*truckSize
            break
    return sum(m)

r=zhuangxiangzi([[5,5],[2,4],[3,1]],6)
print(r)