我正在基于此代码实现背包问题,我想知道最终选择了哪些项。
代码:
Knapsack {
static int max(int a, int b) { return (a > b) ? a : b; }
static int knapSack(int W, int wt[], int val[], int n)
{
int i, w;
int K[][] = new int[n + 1][W + 1];
// Build table K[][] in bottom up manner
for (i = 0; i<= n; i++) {
for (w = 0; w<= W; w++) {
if (i == 0 || w == 0)
K[i][w] = 0;
else if (wt[i - 1]<= w)
K[i][w] = max(val[i - 1] + K[i - 1][w - wt[i - 1]], K[i - 1][w]);
else
K[i][w] = K[i - 1][w];
}
}
return K[n][W];
}
public static void main(String args[])
{
int val[] = new int[] { 60, 100, 120 };
int wt[] = new int[] { 10, 20, 30 };
int W = 50;
int n = val.length;
System.out.println(knapSack(W, wt, val, n));
}
}
不过,我想做的是根据它们最初在数组中的存储方式,找到哪些项最终被插入到背包中,并将这些结果以二进制格式存储在文件中。
我的意思是,在这段代码中有3个项,如果最终选择了1号项(值为60的项)和3号项(值为120的项),我想得到一个类似于1,0,1的字符串
最佳答案
您可以在return语句之前将以下内容添加到代码中:
int ind = n;
int weight = W;
String s = "";
while (ind > 0) {
if (K[ind][weight] != K[ind - 1][weight]) {
s = "1," + s;
weight -= wt[ind - 1];
}
else {
s = "0," + s;
}
ind -= 1;
}
s = s.substring(0, s.length() - 1);
System.out.println(s);
此代码基本上从最终解决方案回溯,以查看i项目的最大值是否与i - 1项(指示是否使用第i项)的最大值相等。
将此添加到代码中后,我将得到以下输出,这似乎是正确的:
0,1,1
220
关于java - 如何知道背包问题(DP实现)中选择了哪个项目?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/56720979/