题解 | #称砝码# 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;
}
全部评论

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
15
1
分享

创作者周榜

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