【2022-11-15每日一题】1710. 卡车上的最大单元数[Easy]

2022-11-15
1分钟阅读时长

2022-11-15每日一题:1710. 卡车上的最大单元数

难度:Easy

标签:贪心 、 数组 、 排序

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

方法一:贪心(排序+遍历)

详细思路过程见官方题解,这里只做个人刷题记录,方便后续查询阅读

func maximumUnits(boxTypes [][]int, truckSize int) int {
    // 按单元个数排序
    sort.Slice(boxTypes, func (i, j int) bool {
        return boxTypes[i][1] > boxTypes[j][1]
    })
    ans := 0
    for _, types := range boxTypes {
        if types[0] >= truckSize {
            ans += truckSize*types[1]
            break
        }
        ans += types[0]*types[1]
        truckSize -= types[0]
    }
    return ans
}

复杂度分析

  • 时间复杂度:O(nlogn),其中 n 是 boxTypes 的长度。排序需要 O(nlogn) 的时间。

  • 空间复杂度:O(logn),其中 n 是 boxTypes 的长度。排序需要 O(logn) 的递归调用栈空间。

LeetCode题库地址