牛客-华为机试-中等-字符串 哈希-HJ41称砝码-C++

描述

现有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五种重量。   

代码

#include<iostream>
#include<vector>
#include<unordered_set>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int>weight(n,0);
    vector<int>num(n,0);
    for(int i=0;i<n;i++)
        cin>>weight[i];
    for(int i=0;i<n;i++)
        cin>>num[i];
    unordered_set<int> s;
    s.insert(0);
    for(int i=0;i<n;i++){
        for(int j=1;j<=num[i];j++){
            unordered_set<int>temp(s);
            for(auto iter=temp.begin();iter!=temp.end();iter++){
                s.insert(*iter+weight[i]);
            }
        }
    }
    cout<<s.size()<<endl;
    return 0;
}