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