现有编号从 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
int maximumBags(vector<int>& capacity, vector<int>& rocks, int additionalRocks) {
vector<int> needs;
long int needsTotal = 0;
for (size_t i = 0; i < capacity.size(); i++){
needs.push_back(capacity[i] - rocks[i]);
needsTotal += needs[i];
}
//总需求量小于供给量
if (needsTotal<=additionalRocks){
return capacity.size();
}
sort(needs.begin(), needs.end());
for (size_t i = 0; i< capacity.size(); i++){
if (needs[i]!=0&&needs[i]<= additionalRocks){
additionalRocks -= needs[i];
needs[i] = 0;
}
}
int maxnum = count_if(needs.begin(), needs.end(), [](int a) {return a == 0; });
return maxnum;
}
注意:考虑变量定义的类型是否会溢出,例如:long int needsTotal;