多重部分和问题

有n种大小不同的数字,各种各mi个,判断是否可以从这些数字之中选出若干个使他们的和恰好为K。
输入:
第一行为n
第二行为n种数字
第三行为这n中数的个数
第四行为K

输出:
可以 输出”yes”
不可以 输出“no”

/*
3
3 5 8
3 2 2
17
*/

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

int dp[100][10000];//前i+1种数加和为j后,剩余的第i+1中数的个数
int main(){
    int n,a[100],b[100],W;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        cin>>b[i];
    }
    cin>>W;
    for(int j=0;j<=W;j++){
        if(j%a[0]==0){
            dp[0][j]=b[0]-(j/a[0]);
        }
        else dp[0][j]=-1;
    }
    for(int i=1;i<n;i++){
        for(int j=0;j<=W;j++){
            if(dp[i-1][j]>=0) dp[i][j]=b[i];
            else if(j<a[i]||dp[i][j-a[i]]<=0) dp[i][j]=-1;
            else dp[i][j]=dp[i][j-a[i]]-1;
        }
    }
    int w=W;
    /*
    cout<<dp[n-1][W]<<" ";
    for(int i=n-1;i>0;i--){
        w=w-(b[i]-dp[i][w])*a[i];
        cout<<dp[i-1][w]<<" ";
    }*/
    if(dp[n-1][W]>0) cout<<"yes";
    else cout<<"no";
    return 0;
}

重复利用数组,改进后:

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

int dp[10000];//加和为j后剩余的a[i]的个数 
int main(){
    int n,a[100],b[100],W;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        cin>>b[i];
    }
    cin>>W;
    memset(dp,-1,sizeof(dp));
    dp[0]=b[0];
    for(int i=0;i<n;i++){
        for(int j=0;j<=W;j++){
            if(dp[j]>=0) dp[j]=b[i];
            else if(j<a[i]||dp[j-a[i]]<=0) dp[j]=-1;
            else dp[j]=dp[j-a[i]]-1;
        } 
    }
    if(dp[W]>=0) cout<<"yes";
    else cout<<"no"; 
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值