AcWing 11. 背包问题求方案数

AcWing 11. 背包问题求方案数

有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。

第 i 件物品的体积是 vi，价值是 wi。

求解将哪些物品装入背包，可使这些物品的总体积不超过背包容量，且总价值最大。

输出 最优选法的方案数。注意答案可能很大，请输出答案模 10^9 + 7 的结果。

输入格式

第一行两个整数，N，V，用空格隔开，分别表示物品数量和背包容积。

接下来有 N 行，每行两个整数 vi,wi，用空格隔开，分别表示第 i 件物品的体积和价值。

输出格式

输出一个整数，表示方案数模10^9+7的结果。

数据范围

0<N,V≤1000
0<vi,wi≤1000

输入样例：

4 5
1 2
2 4
3 4
4 6

输出样例：

2

题解：

用f[i]表示体积为 i 时的最优解，状态g[i]表示当体积为 i 时的方案数(初始化g[0] = 1)，在得出每个状态的最优解时，判断最优解是由f[j]转移过来的还是f[j - v]转移过来的：

1. 如果是f[i]转移过来的，g[i] = g[i];
2. 如果是f[i - v]转移过来的，g[i] = g[i - v];
3. 如果选择f[i]和f[i - v]都是最优解，g[i] = g[i] + g[i - v];

注意在求解最优解时，为了防止重复计算，本题相当于求解背包体积恰好装满时的最优解，需要将f[0]置0，f[1 ~ m] = -INF，否则相同方案会因为背包体积的变化而重复计算。

最后输出g[m]的值就代表体积为m时的方案数，本题方案数很大，计算过程中要对1e9 + 7取模。

AC代码:

#include <iostream>
using namespace std;
const int N = 1010, mod = 1e9 + 7, INF = 0x3f3f3f3f;
int n, m;
int f[N], g[N];
int main() {
	cin >> n >> m; 
	g[0] = 1;
	for (int i = 1; i <= m; i ++) f[i] = -INF; // 初始化-INF, 求解背包恰好装满的情况，防止重复统计
	for (int i = 0; i < n; i ++) { // 枚举前i个物品
		int v, w;
		cin >> v >> w;
		for (int j = m; j >= v; j --) { // 体积从大到小枚举
			int t = max(f[j], f[j - v] + w);
			int s = 0;
			if (t == f[j]) s += g[j]; // 判断是哪种方案转移过来的 加上方案数
			if (t == f[j - v] + w) s += g[j - v];
			s %= mod; // 取模
			f[j] = t;
			g[j] = s;
		}
	}
	int maxw = 0;
	for (int i = 0; i <= m; i ++) maxw = max(maxw, f[i]); // 记录最大值为最优解
	int res = 0;
	for (int i = 0; i <= m; i ++) {
		if (maxw == f[i]) {
			res += g[i]; // 加上满足条件的方案数
			res %= mod; // 取模
 		}
	}
	cout << res << '\n';
	return 0;
}