多重部分和问题

有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;
}