多重部分和问题

有n种不同大小的数字a[i]，每种各m[i]个。判断是否可以从这些数字中选出若干使它们的和恰好为K。

限制条件:1<=n<=100,1<=a[i]，m[i]<=100000,1<=K<<100000

这个问题可以用DP求解，如何定义递推式影响最后的时间复杂度。

定义dp[i+1][j]：用前i+1种数字(数字的编号是从0到i)是否能加成j

为了用前i+1种数字加成j，就需要能前i种数字加成j，j-a[i]，……，j-m[i]*a[i]的一种。

因此有以下递推式:dp[i+1][j]=(0<=k<<m[i]&&k*a[i]<=j时存在使dp[i][j-k*a[i]]为真的k。

可以写出如下代码:

int n;
int K;
int a[100005];
int m[100005];
bool dp[105][100005];
void solve()
{
                dp[0][0]=true;
                for(int i=0;i<n;i++){
                                for(int j=0;j<=K;j++){
                                                for(int k=0;k<=m[i]&&k*a[i]<=j;k++){
                                                                dp[i+1][j]|=dp[i][j-k*a[i]];//a|=b等价于a=a|b，即或运算
                                                }
                                }
                }
                if(dp[n][K]) cout<<"YES"<<endl;
                else cout<<"NO"<<endl;
}

这个算法的时间复杂度为O(K∑m[i])，这样并不够好。一般用DP求取bool结果的话会有不少浪费。

定义dp[i+1][j]：用前i+1种数字(编号从0到i)加和得到j时第i种数最多能剩下多少(不能加成j时为-1)，这样可以减少时间复杂度。

按照上述递推式，如果前i+1种数能加成j，第i+1种数可以剩下m[i]。此外，如果前i种数加和出j-a[i]时第i+1种数还剩下k个，那么加和出j时第i+1种数还剩下k-1个。

因此可以写出递推式。

dp[i+1][j]=m[i](dp[i][j]>=0)

dp[i+1][j]=-1(j<a[i]||dp[i][j-a[i]]<=0)

dp[i+1][j]=dp[i][j-a[i]](其他情况)

最后只需判断dp[n][K]>=0是否成立即可。

这个递推式可以在O(nK)的时间内算出结果，同时考虑数组重复利用，<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">可写出如下代码:</span>

int n;
int K;
int a[100005];
int m[100005];
int dp[100005];
void solve()
{
                dp[0]=0;
                for(int i=0;i<n;i++){
                                for(int j=0;j<=K;j++){
                                                if(dp[j]>=0) dp[j]=m[i];
                                                else if(j<a[i]||dp[j-a[i]]<=0) dp[j]=-1;
                                                else dp[j]=dp[j-a[i]]-1;
                                }
                }
                if(dp[K]>=0) cout<<"YES"<<endl;
                else cout<<"NO"<<endl;
}