HJ41 称砝码

描述

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

注：

称重重量包括 0

数据范围：每组输入数据满足 1≤n≤10  ，1≤mi​≤2000  ， 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五种重量。   

 AC code

/// HJ41 称砝码
#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    while(cin>>n)
    {
        vector<int> weight(n);
        for(int i=0; i<n; i++)
        {
            cin>>weight[i];
        }
        vector<int> num(n);
        for(int i=0; i<n; i++)
        {
            cin>>num[i];
        }

        set<int> kinds;
        kinds.insert(0);
        for(int i=0; i<weight.size(); i++)
        {
            for(int j=0 ; j<num[i]; j++)
            {
                set<int> temp(kinds);
                for(auto it = temp.begin(); it!=temp.end(); it++)
                {
                    kinds.insert(*it+weight[i]);
                }
            }
        }

        cout<<kinds.size()<<endl;
    }
}