分枝界定法求解0-1背包问题python

最新推荐文章于 2023-12-13 17:12:59 发布

godli_one

最新推荐文章于 2023-12-13 17:12:59 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： python 程序人生

本文链接：https://blog.youkuaiyun.com/qq_24059779/article/details/85882149

程序人生同时被 2 个专栏收录

12 篇文章

订阅专栏

python

10 篇文章

订阅专栏

本文介绍了一种使用二叉树和队列数据结构解决背包问题的算法。通过递归地构建二叉树，算法评估了所有可能的物品组合，以找到在给定容量限制下价值最大的物品集合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#coding : utf-8
import numpy as np
import queue
import math
w = [3,5,2,1]#weight
v = [9,10,7,4]#value
N=4
def test(capacity):
    vec_len = 2**(len(v)+1) - 1#tree `s size
    vec_v = np.zeros(vec_len)
    vec_w = np.zeros(vec_len)
    vec_w[0]=capacity
    que = queue.Queue();
    que.put(0)
    best = 0
    while(not que.empty()):
        current = que.get()
        level = int(math.log(current+1,2))
        if(vec_v[current] > vec_v[best]):
            best = current
 
        left = 2*current+1#left child   index
        right = 2*current+2#right child index
 
        if(left < vec_len and vec_w[current]-w[level] >= 0 ):
            vec_v[left] = int(vec_v[current]+v[level])
            vec_w[left] = vec_w[current]-w[level]
            que.put(left)
        if(right < vec_len and sum(v[level+1:N])+vec_v[current] >= vec_v[best]):
            vec_v[right] = vec_v[current]
            vec_w[right] = vec_w[current]
            que.put(right)
    print vec_w[best],vec_v[best]
 
if  __name__ == '__main__':
    test(7)