题目链接: POJ 1742 Coins
题目大意:
有n种不同面值的硬币 A1,A2,...,An , 分别有 C1,C2,...,Cn 枚。
求这些硬币所能组成的小于等于m的不同的面值种类数。
输入格式:
多组输入, 每组2行。
第一行, n,m
第二行 A1,...,An,C1,...,Cn
当 n=0 且 m=0 时表示输入结束
输出格式:
每组一行, 输出一个数字, 表示所能组成的面值之和的种类数
题解:
根据《背包问题九讲》, 该题为多重背包问题。
在理解了“0-1背包问题” 和 “完全背包问题”之后再来理解这个“多重背包问题”, 会觉得轻松许多。
首先想到方法是 定义 dp[n][m] 为 bool 数组, dp[i][j]=true 表示前i种硬币可以组合成面值之和为 j ,dp[i][j]=false 则表示不能。
dp[i][j]=true , 则必须存在一个 k ,使得dp[i−1][j−k∗A[i]]=true,0<=k<=C[i] 。
最后再数一下 dp[n][1]到dp[n][m] 中 true 的个数, 就是答案。
dp[0][0] = true; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { for (int k = 0; k <= C[i]; k++) { dp[i][j] |= dp[i-1][j-k*A[i]]; } } }
复杂度为 O(m∑ni=1ci) ,挺高的了。
在完全背包问题中, 也有类似的结构, 可以通过 dp[i][j]=max(dp[i−1][j],dp[i][j−w[i]]+v[j]) 来避免第三层 k 的循环中做的一些重复计算。但是完全背包问题的公式并不能照搬过来。因为完全背包问题中, 每件物品的个数是无限的, 所以在计算dp[i][j] 的时候可以利用 dp[i][j−w[i]] 。但是在的多重背包问题中, 每件物品的个数是有限制的, 有可能在 dp[i][j−w[i]] 中, 已经用完了第i种物品,如果在计算 dp[i][j] 的时候没有捕捉到这个信息, 就会出错。参考了《挑战程序设计竞赛》P63的“多重部分和问题”的算法:
dp[i][j] 表示使用前 i 种硬币,面值之和达到j 时, 第 i 种硬币最多还剩余多少个,不能达到j ,则为-1。
如果 dp[i−1][j]>=0 ,说明前 i−1 种硬币已经可以达到总面值为 j ,所以dp[i][j]=C[i] 。
此外, 当面值之和为 j−A[i] 时第 i 种硬币还剩k 个的话, 则用这i种硬币使得面值之和为 j 时, 第i 种硬币就能剩下 k−1 个。即 dp[i][j]=dp[i][j−A[i]]−1
dp[i][j]=⎧⎩⎨C[i],−1,dp[i][j−A[i]]−1,if dp[i−1][j]>=0 if j<ai or dp[i][j−A[i]]<=0other
最后, dp[n][1] ~ dp[n][m] 中大于等于0的项的个数, 就是答案。可以参考0-1背包问题,利用滚动数组在空间复杂度上进行优化。不然这题内存会爆。
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#define MAXN 110
#define MAXM 100010
using namespace std;
int A[MAXN], C[MAXN];
int dp[MAXM];
int n, m;
int main() {
while (scanf("%d%d", &n, &m) == 2) {
if (n == 0 && m == 0) break;
memset(A, 0, sizeof(A));
memset(C, 0, sizeof(C));
memset(dp, -1, sizeof(dp));
for (int i = 1; i <= n; i++) scanf("%d", &A[i]);
for (int i = 1; i <= n; i++) scanf("%d", &C[i]);
for (int j = 0; j*A[1] <= m && j <= C[1]; j++)
dp[j*A[1]] = C[1] - j;
for (int i = 2; i <= n; i++) {
for (int j = 0; j <= m; j++) {
if (dp[j] >= 0) dp[j] = C[i];
else if (j < A[i] || dp[j-A[i]] <= 0) dp[j] = -1;
else dp[j] = dp[j-A[i]] - 1;
}
}
int ans = 0;
for (int i = 1; i <= m; i++) {
if (dp[i] >= 0) {
ans++;
}
}
printf("%d\n", ans);
}
return 0;
}