现有编号从 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 个背包装满石头的情况。
注意,不必用完所有的额外石头。
代码
class Solution {
public:
int maximumBags(vector<int>& capacity, vector<int>& rocks, int additionalRocks) {
int n = capacity.size();
vector<int> needed(n);
int count = 0;
for(int i = 0; i < n; i++){
needed[i] = capacity[i] - rocks[i];
}
sort(needed.begin(), needed.end());
for(int &sb : needed){
if(additionalRocks >= sb){
additionalRocks -= sb;
count++;
}
else
{
break;
}
}
return count;
}
};
这道题首先题目给出了你capacity和rocks两个动态数组vector,这是比较常见的贪心题,我们首先先计算背包最大容量和背包现有的石头的差,计算出背包剩余空间。因为要填满最多背包,肯定要从剩余空间少的开始填,所以将每个背包的剩余其进行升序排列。如果额外石头数量大于背包石头剩余空间,就将它填满,否则,由于已经进行过升序排序,说明后面背包都无法填满,退出循环。