0-1 knapsack dynamic programming by Python

import os,sys

class knapsack01_dp:
    def __init__(self, w, v, C):
        self.w =w
        self.v = v
        self.C = C

    def solve(self):
        assert len(self.w) == len(self.v)
        if len(self.w) <= 0:
            return 0
        if self.C <= 0:
            return 0
        n = len(self.w)
        if n <= 0:
            return 0
        self.mem = []
        for i in range(len(w)):
            tmp_l = [0 for _ in range(C+1)]
            self.mem.append(tmp_l)
        for i in range(C+1):
            self.mem[0][i] = v[0] if i >= w[0] else 0
        for i in range(1, n):
            for j in range(self.C+1):
                self.mem[i][j] = self.mem[i-1][j]
                if j >= w[i]:
                    self.mem[i][j] = max(self.mem[i][j], self.mem[i-1][j-w[i]]+v[i])

        print("self.mem: {}".format(self.mem))
        return self.mem[n-1][self.C]

if __name__ == "__main__":
    w = list(range(20)) # each weight for 20 objs
    v = list(range(1, 101, 5)) # each value for 20 objs
    C = 200 # Capacity limitation
    k = knapsack01_dp(w, v, C) # Initlize a knapsack01 solver object
    max_v = k.solve() # get the max vale under given optimization constrains
    print("max_v: {}".format(max_v))    
    max_v = k.solve_opt()
    print("max_v: {}".format(max_v))
    

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值