贪心:一道简单题的细节问题
题目链接:2279. 装满石头的背包的最大数量 - 力扣(LeetCode)
思路:计算出每个背包剩余的容量,再贪心即可。
反面教材:
错在第19行和第20行。我们的逻辑是:剩余的额外石头的数量left减去填充进去的石头数量。但这里我们先让capacity[i]变成了0,那么left再自减capacity[i]就不起作用了。
正确代码:
int maximumBags(vector<int>& capacity, vector<int>& rocks, int additionalRocks)
{
int left = additionalRocks;
for (int i = 0; i < capacity.size(); i++)
{
capacity[i] -= rocks[i];//得到每个背包剩余的容量
}
int ans = 0;
sort(capacity.begin(), capacity.end());
for (int i = 0; i < capacity.size(); i++)
{
if (left == 0 || left < capacity[i]) break;
//如果没有额外的石头,或者额外的石头不足以填满接下来的背包,那么break(因为已排序)
if (capacity[i] == 0) ans++;//可能有些剩余的容量已经是0了
else if (left >= capacity[i])
{
left -= capacity[i];
capacity[i] = 0;
ans++;//处理过的,直接计入答案
}
}
return ans;
}
这里我们不仅处理清楚了逻辑,还进行了优化。实际上,没有必要再遍历一次capacity数组找其中有多少个0了,因为答案=我们填充过的背包数量+原本就没有剩余容量的背包数量。因此,我们在遍历的过程中,直接处理ans即可。