描述
现有n种砝码,重量互不相等,分别为 m1,m2,m3…mn ;
每种砝码对应的数量为 x1,x2,x3...xn 。现在要用这些砝码去称物体的重量(放在同一侧),问能称出多少种不同的重量。
注:
称重重量包括 0
数据范围:每组输入数据满足 1≤n≤10 1≤n≤10 , 1≤mi≤2000 1≤mi≤2000 , 1≤xi≤10 1≤xi≤10
输入描述:
对于每组测试数据:
第一行:n --- 砝码的种数(范围[1,10])
第二行:m1 m2 m3 ... mn --- 每种砝码的重量(范围[1,2000])
第三行:x1 x2 x3 .... xn --- 每种砝码对应的数量(范围[1,10])
输出描述:
利用给定的砝码可以称出的不同的重量数
示例1
输入:
2 1 2 2 1
复制输出:
5
复制说明:
可以表示出0,1,2,3,4五种重量。
代码如下:
#include <iostream>
#include <set>
#include <unordered_set>
#include <vector>
using namespace std;
const int MAX = 2000*10*10+1;
int m[10];
vector<bool> available(MAX);
vector<int> weights;
int main() {
int n;
cin >> n;
weights.emplace_back(0);
weights.reserve(MAX);
for(int i=0;i<n;i++) cin >> m[i];
for(int i=0;i<n;i++) {
int x;
cin >> x;
for(int j=weights.size()-1; j>=0; j--) {
int curr = weights[j];
for(int k=0; k<x; k++) {
curr += m[i];
if(!available[curr]) {
available[curr] = true;
weights.emplace_back(curr);
}
}
}
}
// for(auto x: available) cout << x << endl;
cout << weights.size() << endl;
}
问题:
这里的 for(int j=weights.size()-1;j>=0;j--)为什么不能改成 for(int j=0;j<=weights.size()-1;j++)?
答:当你在循环中添加新的元素到
weights
中时,weights.size()
的值会在每次循环迭代时增加。如果你在循环中使用for(int j=0;j<=weights.size()-1;j++)
这样的方式,循环的终止条件是动态变化的,可能导致循环不断进行,直到j
超出weights
的当前大小。而由后向前遍历的条件则是静态的,不会导致循环不断进行。在循环的每一次迭代中,
weights.size() - 1
的值是不断变化的,因为你在循环内添加了新的元素。如果你在循环开始时weights.size()
为5
,那么第一次迭代时,j
会取值0
,weights.size() - 1
为4
,这样循环会正常运行。但在循环体内部,你向weights
中添加了一个新的元素,此时weights.size()
变成了6
,但是循环条件变成了j <= 5
,所以j
会增加到5
,然后下一次迭代时j
就变成了6
,超出了weights
的范围,循环终止。所以,为了避免这种问题,你应该使用从
weights
的末尾向前遍历的方式,确保你能够在循环中安全地添加新的元素而不会改变当前循环的迭代次数。