题解 | #称砝码#

称砝码

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

题目的主要信息:

  • 现有一组砝码,重量互不相等,分别为 m1m_1,m2m_2,m3m_3mnm_n
  • 每种砝码对应的数量为x1x_1,x2x_2,x3x_3...xnx_n

求解这些砝码能组合出多少种不同的重量。需要注意的是,称重重量包括 0 。

方法一:

采用动态规划。用weights储存不同的砝码重量,nums储存不同重量砝码的数量。maxw表示所有砝码加起来的重量,dp为动态数组,dp[i]=1表示重量i可以用砝码组合出来,dp[i]=0表示重量i不能用砝码组合出来。

用三层for循环对dp数组进行更新,第一层对所有不同重量的砝码进行遍历,第二层对相同重量的砝码个数进行遍历,第三层对重量进行遍历,如果dp[k-weights[i]]==1,就说明不用到当前的砝码可以组成重量k-weights[i],加上当前遍历的砝码可以组成重量k,因此置dp[k]=1。最后遍历一遍dp数组,统计有多少个值为1,则是砝码能组成的重量个数。

具体做法:

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

int main(){
    int n;
    while(cin>>n){
        vector<int> weights;
        int tmp;
        for(int i=0; i<n; i++){//输入砝码的重量
            cin>>tmp;
            weights.push_back(tmp);
        }
        vector<int> nums;
        for(int i=0; i<n; i++){//输入每个砝码的数量
            cin>>tmp;
            nums.push_back(tmp);
        }
        int maxw = 0;//最大重量
        for(int i=0; i<n; i++){//最大重量为所有砝码重量之和
            maxw += nums[i]*weights[i];
        }
        
        vector<int> dp(maxw+1,0);
        dp[0]=1;//初始化为0
        //动态规划
        for(int i=0; i<n; i++){//遍历每个不同重量的砝码
            for(int j=0; j<nums[i]; j++){//遍历每个砝码的数量
                for(int k=maxw; k>=weights[i]; k--){//遍历所有重量
                    if(dp[k-weights[i]]){//如果k-weights[i]重量可以组成的话,那k重量也可以组成的
                        dp[k]=1;
                    }
                }
            }
        }
        int res = 0;
        for(int i=0; i<=maxw; i++){
            if(dp[i]==1){
                res++;
            }
        }
        cout<<res<<endl;
    }
    
    return 0;
}

复杂度分析:

  • 时间复杂度:O(n)O(n),砝码的重量和数量是有限制的,相当于常数级别,因此三重循环的时间复杂度为O(n)O(n)
  • 空间复杂度:O(n)O(n),借用了dp数组的空间。

方法二:

用集合的方法。vec中储存所有砝码,包括相同数量的砝码,用集合s保存所有可能组成的重量。遍历一遍所有砝码,把它加到集合中已组成出的重量中,组合出新的重量,根据集合的性质,若新组合出的重量在集合中没有出现,该重量会更新到s集合中。最后s集合的大小即为可以组成出的重量的个数。 alt 具体做法:

#include<bits/stdc++.h>

using namespace std;

int main(){
    int n;
    while(cin>>n){
        vector<int> vec,weights;
        int temp;
        for(int i =0; i< n;i++){
            cin>>temp;
            weights.push_back(temp);//保存砝码重量
        }
        for(int i= 0; i< n;i++){
            cin>>temp;
            for(int j=0; j< temp;j++){
                vec.push_back(weights[i]);//所有砝码,包括相同重量的多个砝码
            }
        }
        set<int> s;//集合中储存所有可以组合出的重量
        s.insert(0); //0也算是一个初始重量
        for(int i=0; i< vec.size();i++){//遍历一遍所有砝码
            set<int> t=s;
            for(auto it=t.begin();it!=t.end();it++){//把砝码重量加到原本可以组成出的重量上,得到新的重量
                s.insert(*it+vec[i]);
            }
        }
        cout<<s.size()<<endl;
    }
    return 0;
}

复杂度分析:

  • 时间复杂度:O(n2)O(n^2),最坏情况下集合大小为O(n)O(n),双层for循环的时间复杂度为O(n2)O(n_2)
  • 空间复杂度:O(n)O(n),最坏情况下集合的大小为O(n)O(n)
全部评论

相关推荐

我是秋招面的抖音电商部门,从八月开始给我打电话约面试,当时有加对应的hr微信,前几轮还好,从三面开始就是拖半个月一下在推进,面了三轮以后拖了一个月说再加一个交叉面,面完过了半个多月问hr,hr说她离职了不管我了,我连自己的流程在哪里都不清楚。在官网投诉了三次(前两次正常反馈问题,字节你像听不懂话一样),今天下午突然有其他部门约我面试,说在人才库里看见我简历了觉得很匹配,我拒了说不想面了。过了一会,声称原来那个部门的hr负责人给我回电话,表示抱歉,说因为北京hc不够了把我流程释放了,前面的hr是实习生离职了,没有和我对接非常抱歉,如果有想法可以帮我推荐给其他部门。那么我问你:1.&nbsp;从八月等到11月,三个月时间我前面不断确认还有没有hc,给我的回复都是有hc,没hc你面什么呢?2.&nbsp;作为一个大厂你流程这么一坨?让实习生对接校招流程,实习生离职了对应的候选人就不管了?不是我主动问我的流程就这样一直被卡?3.&nbsp;还说我面评没问题,帮我推荐其他部门,现在各个大厂都开奖了,你还让我从头开始面?我前面面了你们三个月啊,孩子饿死了开始装得想负责了。4.&nbsp;你们官网的流程反馈对应的人员是理解能力有问题吗,一个问题我投诉三次才能看明白是吗?套话一堆在哪里随便选了个常用语就给我回复了?不想开可以直接把入口关了
奋斗的小码农a:字节流程就是恶心死人,一次字节广告5面挂一次字节抖音电商6面挂,最后那个面试官装都不装了直接分屏聊天还外放语音家里真的私募了
秋招吐槽大会
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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