回溯法-01背包

题目描述

给定一组多个（n）物品，每种物品都有自己的重量（$w_1,w_2,...,w_n$）和价值（$v_1,v_2,...,v_n$），在限定的总重量/总容量（C）内，选择其中若干个（也即每种物品可以选0个或1个），设计选择方案使得物品的总价值最高。

分析

当物品数不多，可以利用回溯法（子集树模板），遍历每一条路径，利用重量限制剪枝，最终选取满足条件的最大值。

代码

class SubSetTree:
    def __init__(self):
        self.a = []      # 数据列表
        self.w = []
        self.n = 0 # 数据长度
        self.sumVal = 0
        self.res = 0
        self.sumWei = 0
        self.C = 0


    # 子集树递归模板
    def backtrack(self, k): 
        if k >= self.n: #遍历到底
            pass
        else:
            for i in [1, 0]: # 遍历元素 a[k] 的两种选择状态:1-选择，0-不选
                if i==1:
                    self.sumVal += self.a[k]
                    self.sumWei += self.w[k]
                if self.sumWei <= self.C: #非剪枝条件
                    self.res = max(self.res, self.sumVal)
                    self.backtrack(k+1)
                if i==1:
                    self.sumVal -= self.a[k]
                    self.sumWei -= self.w[k]


    def Sovle01Bag(self, w, a, C):
        self.w = w
        self.a = a
        self.C = C
        self.n = len(w)
        self.backtrack(0)
        return self.res




if __name__ == '__main__':
    test = SubSetTree()
    w = [2,2,6,5,4]
    v = [6,3,5,4,6]
    res = test.Sovle01Bag(w,v,10)
    print(res)