背包问题就是:有一堆物品,物品有重量(weight)和价值(value)的属性。求给定背包的容量,最大能容纳的物品总价值。
编号 | 重量 | 价值 |
---|---|---|
1 | 2 | 3 |
2 | 3 | 4 |
3 | 4 | 5 |
4 | 5 | 6 |
最大能装的
01背包
每件物品只能装0件或1件。
二维数组解法:
1)定义一个物品数背包容量的二维数组,值为最大能够得到的价值;
2)初始化,i=0或j=0的元素值设为0;
3)定义状态转移方程
//二维数组解法
int[][] dp = new int[5][9]; //前i件,剩余容量j,dp[i][j]代表对于第i件物品,剩余容量为j的情况下,最大能得到的价值
//填充矩阵
for (int i = 1; i < dp.length; i++) {
for (int j = 1; j < dp[i].length; j++) {
if (j < goods[i].weight) {
dp[i][j] = dp[i-1][j]; //跟前一个一样
} else {
//假设装了的话,跟不装的比较
dp[i][j] = Math.max(dp[i-1][j], dp[i-1][j-goods[i].weight] + goods[i].value); //状态转移方程
}
System.out.print(dp[i][j]);
}
System.out.println("");
}
得出结果矩阵:
0 3 3 3 3 3 3 3
0 3 4 4 7 7 7 7
0 3 4 5 7 8 9 9
0 3 4 5 7 8 9 10
最右下得到的就是最优解,下面就要进行回溯,找到哪些物品是被选择的。
由于 9 -> 10,所以是4号物品是装了的,然后分析未装4号之前的情况(j=10-w(4))是7 -> 7,所以3号没有装。依次分析到1号。
//对矩阵进行回溯,找回解决方案
int[] result = new int[5];
int i=4, j=8;
while (i > 0) {
if (dp[i][j] == dp[i-1][j]) {
//相等代表没有装i号物品
result[i] = 0;
} else {
result[i] = 1;
j = j-goods[i].weight;//回退到没有装i号之前的背包容量
}
i--;
}
System.out.println("方案:");
for (int k = 1; k < result.length; k++) {
System.out.print(result[k] + " ");
}