python 解决 0-1 二维背包问题

问题描述：给定n种物品和一背包。物品i的重量是wi，体积是bi,其价值为vi，背包的容量为C，容积为D。问：应该如何选择装入背包种的物品，使得总价值最大？

其实，这就是一维0-1背包问题的推广，二维背包问题，只要我们把二维背包问题解决了，n维背包问题也就解决了。

二维背包问题，只不过是多加了一个体积的限制条件，那么在纬度上我们需要多加一个纬度，用来衡量体积，具体代码如下：

w = [3, 2, 4]
b = [2, 4, 3]
v = [3, 2, 5]
w_most = 7
b_most = 7

def bag_0_1(w, b, v, w_most, b_most):
    bag_num = len(w)
    w.insert(0, 0)
    b.insert(0, 0)
    v.insert(0, 0)
    dp_table = np.zeros((bag_num+1, w_most+1, b_most+1), np.int)
    for i in range(1, bag_num+1):
        for j in range(1, w_most+1):
            for k in range(1, b_most+1):
                if w[i] <= j and b[i] <= k:
                    dp_table[i][j][k] = max(dp_table[i-1][j][k], dp_table[i-1][j-w[i]][k-b[i]] + v[i])
                else:
                    dp_table[i][j][k] = dp_table[i-1][j][k]
    return dp_table
a = bag_0_1(w, b, v, w_most, b_most)
print(a)
print(a.max())

结果如下：