题解 | #称砝码# 0-1背包问题 动态规划

称砝码

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

思路:

先计算可能的最大值sum,砝码总数len 则建立dp[len+1][sum+1] 表示第i号砝码之前能否凑出重量j 注意len不是类别的数量而是砝码的数量 之后就按照0-1背包思路dp,再设置一个unordered_set,当j被成功凑出后便加入集合 最后输出集合的大小(再额外加一个0)

代码:

#include<iostream>
#include<bits/stdc++.h>

using namespace std;

int main(){
    int N;
    while(cin>>N){
        vector<int> weight;
        vector<int> mount;
        int temp;
        for(int i=0;i<N;i++){
            cin>>temp;
            weight.emplace_back(temp);
        }
        for(int i=0;i<N;i++){
            cin>>temp;
            mount.emplace_back(temp);
        }
        int sum=0;
        for(int i=0;i<N;i++){
            sum=sum+weight[i]*mount[i];
        }
        vector<int> Weight;
        for(int i=0;i<N;i++){
            for(int j=0;j<mount[i];j++){
                Weight.emplace_back(weight[i]);
            }
        }
        int len=Weight.size();
        unordered_set<int> ok;
        vector<vector<bool>> dp(len+1,vector<bool>(sum+1, false));
        for(int i=0;i<len+1;i++){
            dp[i][0]=true;
        }
        for(int i=1;i<len+1;i++){
            for(int j=1;j<sum+1;j++){
                if(Weight[i-1]>j){
                    dp[i][j]=dp[i-1][j];
                }
                else{
                    dp[i][j]=dp[i-1][j]||dp[i-1][j-Weight[i-1]];
                }
                if(dp[i][j]){
                     ok.insert(j);
                }
            }
        }
        
        cout<<ok.size()+1<<endl;
    }
    return 0;
}
全部评论

相关推荐

不愿透露姓名的神秘牛友
07-10 14:00
林子大了什么鸟都有啊,我觉得我说的已经很客气了,阴阳谁呢
牛客62656195...:应该不是阴阳吧?你第一次注册的时候boss就说你是牛人
点赞 评论 收藏
分享
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 11:24
大家还是用ai改吧,我心疼得要死,就当花钱买教训吧,人家直接拿完钱就跑路了
程序员小白条:简历修改700....神奇,又不是帮你面试,咋的,简历修改从双非变92了还是没实习变成有大厂实习了
点赞 评论 收藏
分享
评论
15
1
分享

创作者周榜

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