greedy knapsack code example
Example 1: greedy knapsack
def greedy_knapsack(values,weights,capacity):
n = len(values)
def score(i) : return values[i]/weights[i]
items = sorted(range(n) , key=score , reverse = True)
sel, value,weight = [],0,0
for i in items:
if weight +weights[i] <= capacity:
sel += [i]
weight += weights[i]
value += values [i]
return sel, value, weight
weights = [4,9,10,20,2,1]
values = [400,1800,3500,4000,1000,200]
capacity = 20
print(greedy_knapsack(values,weights,capacity))
Example 2: knapsack algorithm in python
def knapSack(W, wt, val, n):
K = [[0 for x in range(W + 1)] for x in range(n + 1)]
for i in range(n + 1):
for w in range(W + 1):
if i == 0 or w == 0:
K[i][w] = 0
elif wt[i-1] <= w:
K[i][w] = max(val[i-1] + K[i-1][w-wt[i-1]], K[i-1][w])
else:
K[i][w] = K[i-1][w]
return K[n][W]
val = [50,100,150,200]
wt = [8,16,32,40]
W = 64
n = len(val)
print(knapSack(W, wt, val, n))