0/1背包问题 模板 C++实现

问题：给定n种物品和一个背包，物品i的重量是Weight i，价值为Value i，背包容量是c。对每种物品只有两种选择，装入背包或不装入背包，0/1背包问题（0/1 Knapsack Problem）是如何选择装入背包的物品，使得装入背包中物品的总价值最大。

算法：

1、二维dp

dp[i][j]表示只考虑前 i 件物品，当背包容量为 j 时的最大价值。

状态转移方程：对于每个物品 i ，我们有两种选择：不放入背包，或者放入背包。如果不放入背包，那么 dp[ i ][ j ] = dp[ i - 1 ][ j ] ；如果放入背包，前提是背包的容量 j 大于等于物品 i 的重量weight[ i ] ，那么 dp[ i ][ j ] = max(dp[ i - 1 ][ j ] , dp[ i - 1 ][ j - weight[ i ] ] + value[ i ]) 。选择这两种情况的最大值作为 dp[ i ][ j ] 的值。

循环遍历： 当前背包容量为 i 时所得最大价值，必不小于背包容量为 i - 1 时的最大价值，故先令此时背包容量的最大价值为 dp[ i - 1 ][ j ] ，再判断是否能放的下每个不同的物品，如果能放下，则比较后取较大的值。

二维 dp 对比一维 dp ，好在它可以通过回溯知道背包中有哪些物品。如果 dp[ i ][ j ] > dp[ i - 1 ][ j ] ,证明这个物品被装入了，反之则没有。

回溯代码如下：

for(i=n,j=c;i>0;i--){//回溯 如果value比上一个大，说明这件物品放入背包了，置1，如果没大说明没放入，置0
	if(dp[i][j] > dp[i-1][j]){
		x[i] = 1;
		j -= weight[i];
	}
	else x[i] = 0;
}

代码：

int main() {
	// 背包容量 = 9 ，物品个数 = 4。要求不超c，求最大value。
	int weight[4] = { 1,3,4,5 }, value[4] = { 2,4,7,3 }, n = 4, c = 9;
	int dp[5][10];
	memset(dp, 0, sizeof(dp));// 初始化 dp
	for (int i = 1; i <= n; i++)
		for (int j = 0; j <= c; j++) {
			dp[i][j] = dp[i - 1][j];// 不拿这个物品
			if (j >= weight[i - 1])	dp[i][j] = max(dp[i][j], dp[i - 1][j - weight[i - 1]] + value[i - 1]);// 可以拿
		}
	cout<<dp[n][c]<< endl;
	return 0;
}

2、一维dp

dp[j]表示背包容量为 j 时，能装入的最大价值。

使用 一维dp 时，要注意需要逆序遍历背包容量，因为 一维dp 无法存储物品下标，即无法判断某个物品是否被装入。如果正序遍历背包容量，那么可能会出现重复装入某个物品的情况。

代码：

int main() {
	int weight[4] = { 1,3,4,5 }, value[4] = { 2,4,7,3 }, n = 4, c = 9;
	int dp[10] = { 0 };
	for (int i = 0; i < n; i++)// 遍历物品
		for (int j = c; j >= 0; j--)// 逆序遍历背包容量
			if (j > weight[i])	dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
	cout << dp[c] << endl;
	return 0;
}