01 背包问题
原题链接
有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。
第 i 件物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品数量和背包容积。
接下来有 N 行,每行两个整数 vi,wi,用空格隔开,分别表示第 i 件物品的体积和价值。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤1000
0<vi,wi≤1000
输入样例
4 5
1 2
2 4
3 4
4 5
输出样例:
8
#include <iostream>
using namespace std;
int dp[1005][1005];
// dp[i][j]:第i个物品,在体积为j时的最大价值
// dp[i][j] = dp[i-1][j] // 第i个物品的最大价值==第i-1个物品的最大价值
// dp[i][j] = max(dp[i][j], dp[i-1][j-v[i]]+w[i]) // 第i个物品的最大价值==当前物品的最大价值和第i-1个物品体积为j-v[i]时装入第i个物品的最大价值
int main() {
int N,V;
cin >> N >> V;
int v[N+1],w[N+1];
for(int i = 1; i <= N; i++) {
cin >> v[i] >> w[i];
}
for(int i = 1; i <= N; i++) {
for(int j = 0; j <= V; j++) {
dp[i][j] = dp[i-1][j];
if(j >= v[i]) {//体积j>=v[i]能够装下第i个物品
dp[i][j] = max(dp[i][j],dp[i-1][j-v[i]]+w[i]);
}
}
}
cout <<dp[N][V] << endl;//放入N个物品,体积为V时的最大价值
return 0;
}
开一个dp数组,dp[i]:表示占用i个空间大小的背包容量时,背包中的物品价值最大为多少。
这里的dp转换逻辑是:dp[j] = max(dp[j],dp[j-v[i]]+w[i]); j-v[i]的物品的价值+当前物品的价值,与原来的 j 容量背包物品的价值取最大值。
#include <iostream>
using namespace std;
// 优化:dp[j]: 体积为j时的最大价值,
//若循环从0~V的话,则会造成重复加入物品,而01背包要求一个物品只能加入一次,于是我们选择从V~v[i]的for循环
int dp[1005];
int main() {
int N,V;
cin >> N >> V;
int v[N+1],w[N+1];
for(int i = 1; i <= N; i++) {
cin >> v[i] >> w[i];
}
for(int i = 1; i <= N; i++) {
for(int j = V; j >= v[i]; j--) {
dp[j] = max(dp[j-v[i]]+w[i],dp[j]);
}
}
cout <<dp[V] << endl;
return 0;
}