题解 | #称砝码#

称砝码

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

分享此类多重背包问题的二维dp数组解法和一维dp数组解法
//本质上是个多重背包问题
//首先用二维dp数组解决一下
#include<stdio.h>
#include<string.h>
int main(void)
{
    int n;
    scanf("%d",&n);
    int m[15]={0};
    int x[15]={0};
    int sum=0;
    int dp[15][200000]={0};//0表示不可以称出这个重量,1表示可以
    for(int i=1;i<=n;i++){scanf("%d",&m[i]);}
    for(int i=1;i<=n;i++){
        scanf("%d",&x[i]);
        sum+=x[i]*m[i];//计算可以称出的最大重量
    }
    //开始动态规划
    for(int i=0;i<=n;i++){dp[i][0]=1;}//dp数组初始化
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=sum;j++)
        {
            if(dp[i-1][j]==1)
            {
                dp[i][j]=1;
            }
            else
            {
                for(int k=1;k<=x[i];k++)
                {
                    if(j>=k*m[i]&&dp[i-1][j-k*m[i]]==1)
                    {
                        dp[i][j]=1;
                        break;
                    }
                }
            }
        }
    }
    int t=0;
    for(int i=0;i<=sum;i++)
    {
        t+=dp[n][i];
    }
    printf("%d\n",t); 
    return 0;
}
//本质上是个多重背包问题
//然后用滚动dp数组解法解决一下
#include<stdio.h>
#include<string.h>
int main(void)
{
    int n;
    scanf("%d",&n);
    int m[15]={0};
    int x[15]={0};
    int sum=0;
    int dp[200000]={0};//0表示不可以称出这个重量,1表示可以
    dp[0]=1;
    for(int i=1;i<=n;i++){scanf("%d",&m[i]);}
    for(int i=1;i<=n;i++){
        scanf("%d",&x[i]);
        sum+=x[i]*m[i];//计算可以称出的最大重量
    }
    //开始动态规划
    for(int i=0;i<=n;i++)
    {
        for(int j=sum;j>=0;j--)
        {
            if(dp[j]==1){dp[j]=1;continue;}
            for(int k=1;k<=x[i];k++)
            {
              if(k*m[i]<=j&&dp[j-k*m[i]]==1)
              {
                  dp[j]=1;
                  break;
              }
            }
            
        }
    }
    int t=0;
    for(int i=0;i<=sum;i++)
    {
        t+=dp[i];
    }
    printf("%d",t);
    return 0;
}



全部评论

相关推荐

牛客771574427号:恭喜你,华杰
点赞 评论 收藏
分享
11-05 07:29
贵州大学 Java
点赞 评论 收藏
分享
11-27 12:43
已编辑
门头沟学院 C++
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务