在背包 0/1 代碼中實現回溯表

Question

我正在嘗試在我的動態編程算法中為背包 0/1 問題實現一個回溯表。 在我的代碼中，我設置了一個回溯表矩陣，但我不確定如何填寫關於我的程序的這些值。

def optimizeInvestments(invstmt, money):
""" knapsack problem """
n = len(invstmt)
val = []
name = []
roi = []
traceback = [[0 for i in range(n)] for i in range(n)]

for i in invstmt:
    name.append(i[0])
    val.append(i[-1])
    roi.append(i[1])

K = [[0 for x in range(money + 1)] for x in range(n + 1)]
I = [[0 for x in range(money + 1)] for x in range(n + 1)]

for i in range(n + 1):
    for w in range(money + 1):
        if i == 0 or w == 0:
            K[i][w] = 0
            I[i][w] = ""

        elif roi[i - 1] <= w:

            if (val[i - 1] + K[i - 1][w - roi[i - 1]] > K[i - 1][w]):
                K[i][w] = val[i - 1] + K[i - 1][w - roi[i - 1]]
                if len(I[i - 1][w - roi[i - 1]]) > 0:
                    I[i][w] = name[i - 1] + " & " + I[i - 1][w - roi[i - 1]]
                else:
                    I[i][w] = name[i - 1]

            else:
                K[i][w] = K[i - 1][w]
                I[i][w] = I[i - 1][w]

        else:
            K[i][w] = K[i - 1][w]
            I[i][w] = I[i - 1][w]

portfolio = 'With $'+ str(money) + ", invest in " + str(I[n][money]) + " for a ROI of $" + str(K[n][money])
return portfolio

Answer 1

弄清楚如何添加回溯表。 這是我的代碼：

def optimizeInvestments(invstmt, money):
""" knapsack problem """
n = len(invstmt)
val = []
name = []
roi = []

for i in invstmt:
    name.append(i[0])
    val.append(i[-1])
    roi.append(i[1])

K = [[0 for x in range(money + 1)] for x in range(n + 1)]
I = [[0 for x in range(money + 1)] for x in range(n + 1)]
optimal = [[None for x in range(money + 1)] for x in range(n + 1)]
traceback = [[False for x in range(money + 1)] for x in range(n + 1)]


for i in range(n + 1):
    for w in range(money + 1):
        optimal[i][w] = 0
        if i == 0 or w == 0:
            K[i][w] = float(0)
            I[i][w] = ""
            optimal[i][w] = 0
            traceback[i][w] = False

        elif roi[i - 1] <= w:

            if (val[i - 1] + K[i - 1][w - roi[i - 1]] > K[i - 1][w]):
                K[i][w] = val[i - 1] + K[i - 1][w - roi[i - 1]]
                optimal[i][w] = val[i - 1] + optimal[i - 1][w - roi[i - 1]]
                traceback[i][w] = True

                if len(I[i - 1][w - roi[i - 1]]) > 0:
                    I[i][w] = name[i - 1] + " & " + I[i - 1][w - roi[i - 1]]
                else:
                    I[i][w] = name[i - 1]

            else:
                K[i][w] = K[i - 1][w]
                I[i][w] = I[i - 1][w]
                optimal[i][w] = optimal[i-1][w]
                traceback[i][w] = False




        else:
            K[i][w] = K[i - 1][w]
            I[i][w] = I[i - 1][w]
            optimal[i][w] = optimal[i-1][w]
            traceback[i][w] = False

    print('optimal: ')
    for row in optimal:
        print(row)
    print("traceback: ")
    for row in traceback:
        print(row)