0-1背包问题 python

原理：当前物品重量wi大于当前背包容量w的时候，直接去前一个物品在背包容量为w时的最大价值，否则取前一个物品在w时的最大价值 和 前一个物品在w-wi的最大价值加上当前物品vi的价值 中最大的价值。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

' a test module '

__author__ = 'Zhang Shuai'


def bag(n, c, w, v) -> "n 是个数，c是背包最大容量，w是物品的的重量，v是物品的价值":
    w.insert(0, 0)
    v.insert(0, 0)  # 为了是序号跟res保持同步，因为res是从1开始计算的
    res = [[-1 for i in range(c + 1)] for j in range(n + 1)]
    for j in range(c + 1):
        res[0][j] = 0
    for i in range(1, n + 1):
        for j in range(1, c + 1):

            if j < w[i]:
                res[i][j] = res[i - 1][j]
            else:
                res[i][j] = max((res[i - 1][j], res[i - 1][j - w[i]] + v[i]))
    print(res[1:])
    return res


def show(n, c, w, res):
    print('最大价值为:', res[n][c])
    x = [False for i in range(n)]
    j = c
    for i in range(n, 0, -1):
        if res[i][j] > res[i - 1][j]:
            x[i - 1] = True
            j -= w[i - 1]
    print('选择的物品为:')
    for i in range(n):
        if x[i]:
            print('第', i, '个,')
    print('')


if __name__ == '__main__':
    n = 5
    c = 10
    w = [2, 2, 6, 5, 4]
    v = [6, 3, 5, 4, 6]

    res = bag(n, c, w, v)
    show(n, c, w, res)

---

第二钟方法，能节省内存：
因为你会发现，res[i][j],只跟res[i-1][j]和res[i-1][j-w[i]]有关，即j只跟i1层的小于等于那么j的值有关，如果每一行j从最后更新，那么一维数组就能保存所有的信息。

def bag(n, c, w, v) -> "n 是个数，c是背包最大容量，w是物品的的重量，v是物品的价值":
    w.insert(0, 0)
    v.insert(0, 0)  # 为了是序号跟res保持同步，因为res是从1开始计算的
    res = [0 for j in range(c+1)]
    for i in range(1, n + 1):
        for j in range(c, 0, -1): # j从大到小循环，这样可以使一维数组保存所有信息。
            if j<w[i]:
                res[j] =res[j]
            else:
                res[j] = max(res[j],res[j-w[i]] + v[i])
    print(res[1:])
    return res