LeetCode每日一题 - 卡车上的最大单元数

题目：
请你将一些箱子装在 一辆卡车 上。给你一个二维数组 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

思路：先不论类型，按照最大单元数从大到小排列，再通过truckSize判断每类箱子的个数，如果超过就用truckSize减去当前类。

C语言：

不断加上累加当前最大数量，再使其为0，从大到小遍历。

int maximumUnits(int** boxTypes, int boxTypesSize, int* boxTypesColSize, int truckSize)
{
    int nums=0, ans=0;
    while(nums<truckSize)
    {
        int flag, max=0;
        for(int i=0;i<boxTypesSize;i++)
        {
            if(max<boxTypes[i][1])
            {
                max = boxTypes[i][1];
                flag = i;
            }
        }
        if(nums+boxTypes[flag][0] >= truckSize)
            return (ans+=boxTypes[flag][1]*(truckSize-nums));
        else
        {
            ans+=boxTypes[flag][0]*boxTypes[flag][1];
            nums+=boxTypes[flag][0];
            boxTypes[flag][1]=0;
        }
    }
    return ans;
}

执行用时84ms，内存消耗7.4MB。

Python3：

class Solution:
    def maximumUnits(self, boxTypes: List[List[int]], truckSize: int) -> int:
        boxTypes.sort(key=lambda x: (-x[1],x[0]))
        n = 0
        r = 0
        for i in boxTypes:
            if n+i[0] <= truckSize:
                n += i[0]
                r += i[0]*i[1]
            else:
                r += (truckSize - n)*i[1]
                break
        return r

执行用时48ms，内存消耗15.2MB。