题解 | #称砝码#

称砝码

https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c

#include <iostream>
#include <unordered_set>
using namespace std;
const int N = 110, M = 2e5 + 10;

int n, m, cnt;
int w[N], s[N], we[N];
int main() {
    cin >> n;
    for(int i = 1; i <= n; i ++)
        cin >> w[i];
    for(int i = 1; i <= n; i ++)
        cin >> s[i];

    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= s[i]; j ++)
            we[cnt ++] = w[i]; 
    unordered_set<int> h;
    h.insert(0);    
    for(int i = 0; i < cnt; i ++)
    {
        //每次都只在上一个集合状态中加入新砝码查看是否有新状态产生
        unordered_set<int> tmp(h);
        for(auto it : tmp)  h.insert(it + we[i]);  
    }
    cout << h.size() << endl;

    return 0;
}

全部评论

相关推荐

评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务