AcWing 12. 背包问题求具体方案
有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。
第 i 件物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出 字典序最小的方案。这里的字典序是指:所选物品的编号所构成的序列。物品的编号范围是 1…N。
输入格式
输出一行,包含若干个用空格隔开的整数,表示最优解中所选物品的编号序列,且该编号序列的字典序最小。
物品编号范围是 1…N。
输出格式
输出一个整数,表示方案数模10^9+7的结果。
数据范围
0<N,V≤1000
0<vi,wi≤1000
输入样例:
4 5
1 2
2 4
3 4
4 6
输出样例:
1 4
题解:
我们可以通过倒推的方法来输出最优方案,如果 f [ i − 1 ] [ j − v ] + w f[i - 1][j - v] + w f[i−1][j−v]+w的状态比 f [ i − 1 ] [ j ] f[i - 1][j] f[i−1][j]更优,说明第i个物品是被选中的,相反则说明没有被选中。如果 f [ i − 1 ] [ j − v ] + w f[i - 1][j - v] + w f[i−1][j−v]+w和 f [ i − 1 ] [ j ] f[i - 1][j] f[i−1][j]的价值相同,说明选择和不选择这个物品是两种不同的最优方案。
因为这题需要输出字典序最小的方案,所以我们在打表的时候需要将物品按编号从大到小来打表,这样我们第一个判断的就是编号最小的物品有没有被选中,保证字典序最小。另外能选的物品我们要尽量将它选中,这样可以保证输出的结果是字典序最小的。
注意这题不能将二维数组优化成一维,因为一维数组会覆盖掉之前的方案数据,使用二维数组才可以将所有的方案都保留下来,方便我们判断哪些物品被选中了。
AC代码:
#include <iostream>
using namespace std;
const int N = 1005;
int v[N], w[N], f[N][N];
int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i ++) cin >> v[i] >> w[i];
for (int i = n; i >= 1; i --) { // 从后往前枚举物品,方便字典序最小
for (int j = 0; j <= m; j ++) { // 枚举体积,顺序任意
f[i][j] = f[i + 1][j];
if (j >= v[i]) f[i][j] = max(f[i][j], f[i + 1][j - v[i]] + w[i]);
}
}
int vol = m;
for (int i = 1; i <= n; i ++) { // 判断1 - n物品能不能选
if (f[i][vol] == f[i + 1][vol - v[i]] + w[i] && v[i] <= vol) { // 能选就选,保证字典序最小
cout << i << ' '; // 输出被选中的物品
vol -= v[i];
}
}
return 0;
}