华为OJ：称砝码

题目描述
现有一组砝码，重量互不相等，分别为m1,m2,m3…mn；
每种砝码对应的数量为x1,x2,x3…xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。

注：
称重重量包括0

方法原型：public static int fama(int n, int[] weight, int[] nums)

输入描述:
输入包含多组测试数据。

对于每组测试数据：

第一行：n — 砝码数(范围[1,10])

第二行：m1 m2 m3 … mn — 每个砝码的重量(范围[1,2000])

第三行：x1 x2 x3 …. xn — 每个砝码的数量(范围[1,6])

输出描述:
利用给定的砝码可以称出的不同的重量数

输入例子:
2
1 2
2 1

输出例子:
5

解析：
//称砝码
int main() {
    int n;
    cin >> n;
    int max_wight = 0;
    vector<int> nums(n), wights(n);

    for (int i = 0; i < n; i++)
        cin >> wights[i];
    for (int i = 0; i < n; i++) {
        cin >> nums[i];
        max_wight += wights[i] * nums[i]; //求最大可称重的值
    }

    //set内是可以称重的值，元素互斥单调增
    set<int> s;
    s.insert(max_wight);

    //进行n次循环；
    //对于第i次循环，将set中元素从小到大依次取出，
    //减去k * wights[i]，表示少使用k个质量为wights[i]的砝码
    //也可以作为可称重的值；只要这个值是大于0的即可。
    for (int i = 0; i < n; i++) {
        set<int>::iterator iter = s.begin();
        while (iter != s.end()) {
            for (int k = 1; k <= nums[i] && *iter - k * wights[i] > 0; k++)
                s.insert(*iter - k * wights[i]);
            iter++;
        }
    }
    s.insert(0);
    cout << s.size() << endl;
    return 0;
}