一、题目
现有编号从 0
到 n - 1
的 n
个背包。给你两个下标从 0 开始的整数数组 capacity
和 rocks
。第 i
个背包最大可以装 capacity[i]
块石头,当前已经装了 rocks[i]
块石头。另给你一个整数 additionalRocks
,表示你可以放置的额外石头数量,石头可以往 任意 背包中放置。
请你将额外的石头放入一些背包中,并返回放置后装满石头的背包的 最大 数量。
示例 1:
输入:capacity = [2,3,4,5], rocks = [1,2,4,4], additionalRocks = 2
输出:3
解释:
1 块石头放入背包 0 ,1 块石头放入背包 1 。
每个背包中的石头总数是 [2,3,4,4] 。
背包 0 、背包 1 和 背包 2 都装满石头。
总计 3 个背包装满石头,所以返回 3 。
可以证明不存在超过 3 个背包装满石头的情况。
注意,可能存在其他放置石头的方案同样能够得到 3 这个结果。
示例 2:
输入:capacity = [10,2,2], rocks = [2,2,0], additionalRocks = 100
输出:3
解释:
8 块石头放入背包 0 ,2 块石头放入背包 2 。
每个背包中的石头总数是 [10,2,2] 。
背包 0 、背包 1 和背包 2 都装满石头。
总计 3 个背包装满石头,所以返回 3 。
可以证明不存在超过 3 个背包装满石头的情况。
注意,不必用完所有的额外石头。
提示:
- n == capacity.length == rocks.length
- 1 <= n <= 5 * 104
- 1 <= capacity[i] <= 109
- 0 <= rocks[i] <= capacity[i]
- 1 <= additionalRocks <= 109
二、题解
步骤
- 计算出每个背包的剩余容量(用新数组存储)
- 优先填充剩余容量小的背包(新数组升序排序)
- 遍历剩余容量数组,剩余额外石头数量 = 额外石头数量 - 当前背包容量(相当于用石头填充当前背包)
- 循环停止条件:所有背包填充完了、石头正好用完、石头不够填充当前背包
/**
* @param {number[]} capacity
* @param {number[]} rocks
* @param {number} additionalRocks
* @return {number}
*/
var maximumBags = function (capacity, rocks, additionalRocks) {
// 每个背包的剩余容量
let remainCapacity = capacity.map((val, idx) => (val - rocks[idx]))
// 升序排序
remainCapacity.sort((a, b) => a - b)
for (var i = 0; i < remainCapacity.length; i++){
additionalRocks = additionalRocks - remainCapacity[i]
if (additionalRocks == 0) {
// 石头正好用完
return i+1
}
if (additionalRocks < 0) {
// 石头不够用
return i
}
}
return i
};