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]转移过来的:
- 如果是f[i]转移过来的,g[i] = g[i];
- 如果是f[i - v]转移过来的,g[i] = g[i - v];
- 如果选择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;
}