0-1背包

0-1背包问题|完全背包问题

求取每个位置的最佳策略，得到最终位置的最佳策略

N = int(input("物品的数量:"))
M = int(input("背包的容量:"))
weight = [0 for i in range(M+1)]
value = [0 for i in range(N+1)]
matrix = [[0 for i in range(M+1)] for i in range(N+1)]  #每个位置的最大价值

weight[0] = 0
value[0] = 0
for i in range(1,N+1):
    weight[i] = int(input('输入物体'+str(i)+'的重量:'))
    value[i] = int(input('输入物体'+str(i)+'的价值:'))


def lingyi():  # 0-1背包
    for i in range(0,N+1):    # 左闭右开，在前i个物品
        for j in range(0,M+1):   # 在前j个容量 位置下
            if j < weight[i]:  # 容量j小于物品i的重量
                matrix[i][j] = matrix[i-1][j]  # 保持之前的策略, 此位置的最大价值
            else:
                data1 = matrix[i-1][j]
                data2 = matrix[i-1][j-weight[i]]+value[i]  #加入物品i后，与前i-1个物品和j-weight[i]容量下的最大价值之和
                matrix[i][j] = max(data1,data2)   # 选取最好的那个策略
    print("0-1背包最大价值为:",matrix[N][M])

def wanquan():  # 完全背包
    for i in range(0,N+1):    # 左闭右开，在前i个物品
        for j in range(0,M+1):   # 在前j个容量 位置下
            if j < weight[i]:  # 容量j小于物品i的重量
                matrix[i][j] = matrix[i-1][j]  # 保持之前的策略, 此位置的最大价值
            else:
                data1 = matrix[i-1][j]
                data2 = matrix[i][j-weight[i]]+value[i]  # 物品可以重复，前i-1个改为前i个
                matrix[i][j] = max(data1,data2)
    print("完全背包最大价值为:",matrix[N][M])
lingyi()
wanquan()