HJ41 称砝码,set相关操作

set相关操作,
包括insert
autox 遍历

#include 
using namespace std;

int main() {
    
    int n;
    cin >> n;
    vector<int> weight(n,1);
    vector<int> amount(n,1);
    for(int i = 0; i < weight.size(); i++){
        cin >> weight[i];
    }
    for(int i = 0; i < amount.size(); i++){
        cin >> amount[i];
    }
    // cout << weight[0] << weight[1];
    // cout << amount[0] << amount[1];
//先traver weight 再traverse amount
    // int maxWeight;
    set<int> s;
    s.insert(0);
    // cout << s.size() << endl;
    for(int i = 0; i < weight.size(); i++){
        for(int j = 1; j <= amount[i]; j++){
            set<int> temp(s); //用于保存上一次的set,目的是为了每次在原来的已有重量进行配对
            for(auto& x : temp){
                s.insert(x + weight[i]);
            }
        }
    }
    cout << s.size() << endl;
}
// 64 位输出请用 printf("%lld")

你可能感兴趣的:(算法)