华为机试----称砝码

题目

称砝码

代码

#include<iostream>
#include<set>
#include<vector>
using namespace std;

int main(){
   int n,i,j,num,tsum;
   while(cin>>n){
        int weight[n];
        vector<int> seq;
        set<int> s;
        for(i=0;i<n;i++)
            cin>>weight[i];
        for(i=0;i<n;i++){
            cin>>num;
            for(j=0;j<num;j++)
                seq.push_back(weight[i]);
        }
        s.insert(0);
        for(i=0;i<seq.size();i++){
            set<int> t(s);
            for(auto it=t.begin();it!=t.end();it++)
                s.insert(*it+seq[i]);
        }
        cout<<s.size()<<endl;
   }
}

思路（搬运）

1. 首先根据输入顺序，将砝码用数字序列表示，例如2个1g和1个2g，就用 1 1 2的序列表示；
2. set序列用来表示加入当前砝码之前能产生的重量种类；
3. set初始化为{0}；当第一个1g砝码放入时，则set中需要插入原先set中的所有元素+1g后的结果；即{0, 0+1};
4. 当第二个1g加入时，则set会插入{0+1, 1+1},就变成了{0, 1, 2};
重复上述步骤加入所有砝码；则最后set的大小即为能产生的重量种类。

搬运来源：称砝码的思路