01背包問題python 2.7實現

版權聲明：本文為博主原創文章，轉載請注明轉自 http://www.cnblogs.com/kdxb/p/6140625.html

#!/usr/bin/env python
# -*- coding:utf-8 -*-
class bag():
    def __init__(self,weight,value):
        self.weight = weight
        self.value = value
    def knapsack(self, full_weight):#weight value存數組
        result = [[0 for i in range(full_weight+1)] for i in range(len(self.value)+1)]
        count = len(self.weight)#物品個數
        for n in range(1,count+1):#n當前最大物品個數
            for weight in range(0,full_weight+1):#背包內重量遞增
                if self.weight[n-1]<=weight:#第n個背包的重量為weight[n-1]判斷是否小於允許容量
                    if result[n-1][weight]<(result[n-1][weight-self.weight[n-1]]+self.value[n-1]):
                        #如果當前物品在相同重量情況下價值更高
                        result[n][weight]=result[n-1][weight-self.weight[n-1]]+self.value[n-1]
                    else:
                        result[n][weight]=result[n-1][weight]
                else:
                    result[n][weight] =result[n-1][weight]
        for perrow in result:
            print perrow
        return result
    def find_which(self,full_weight):
        result = self.knapsack(full_weight)
        i= len(result)-1
        j = len(result[i])-1
        while i >=1:
            while j>=1:
                if result[i-1][j]!=result[i][j]:#說明當前行的東西拿了
                    print '第'+ str(i)+'個'
                    j = j -self.weight[i-1]
                    i = i - 1
                    break
                else:
                    i = i -1


def main():
    sort_instance = bag([2,2,6,5,4,1,2,7,5,7,4],[6,3,5,4,6,1,4,7,3,6,1])#重量，價值初始化
    sort_instance.find_which(30)#定義背包總重量

if __name__ =='__main__':
    main()

實現結果：