POJ 1742-Coins [dp 多重背包问题] 《挑战程序设计竞赛》2.3

题目链接： POJ 1742 Coins

题目大意：

有n种不同面值的硬币 $A_1, A_2, ... , A_n$， 分别有$C_1, C_2, ... , C_n$ 枚。
求这些硬币所能组成的小于等于m的不同的面值种类数。

输入格式：

多组输入， 每组2行。
第一行, $n, m$
第二行 $A_1, ... ,A_n, C_1, ... , C_n$
当$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\sum_{i=1}^n c_i)$ ，挺高的了。
在完全背包问题中， 也有类似的结构， 可以通过$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] = \begin{cases} C[i], & \text{if $dp[i-1][j] >= 0$ } \\ -1, & \text{if $j < a_i$ or $dp[i][j-A[i]] <= 0$} \\ dp[i][j-A[i]] - 1, & \text{other} \end{cases}$$
最后， $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;
}