-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
Copy pathKnapsack.py
43 lines (38 loc) · 1.53 KB
/
Knapsack.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# Girlscript Summer Of Code
# 0-1 Knapsack program in Python
"""
What is 0-1 KnapSack program?
Given weights and values of n items, put these items in a knapsack of capacity W to get the maximum total value in the knapsack. In other words, given two integer arrays val[0..n-1] and wt[0..n-1] which represent values and weights associated with n items respectively. Also given an integer W which represents knapsack capacity, find out the maximum value subset of val[] such that sum of the weights of this subset is smaller than or equal to W. You cannot break an item, either pick the complete item, or don’t pick it (0-1 property).
[Refered from www.geeksforgeeks.com]
"""
"""
Variables:
mw-maximum weight
wt[]-Array of weights
val[]-Array of values
n-number of values
"""
def knapSack(maxweight, weights, values, n):
subset = [[0 for x in range(maxweight + 1)] for x in range(n + 1)]
for i in range(n + 1):
for j in range(maxweight + 1):
if i == 0 or j == 0:
subset[i][j] = 0
elif weights[i-1] <= j:
subset[i][j] = max(values[i-1] + subset[i-1][j-weights[i-1]], subset[i-1][j])
else:
subset[i][j] = subset[i-1][j]
return subset[n][maxweight]
# Test Case for the problem :
# Alter the values to and weights to check the result
# Corresponding Values and Weights
values = [30, 20, 60, 80, 10]
weights = [10, 20, 30, 40, 50]
# Maximum Weight
MaxWeight = 80
# Number of Values
n = 5
# Display the Result
print(knapSack(MaxWeight, weights, values, n))
# Result is 170 for the test case given initially.
# Happy Coding!