我正在基于此代码实现背包问题,我想知道最终选择了哪些项。
代码:

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/

10-10 19:47