Python/greedy_methods/fractional_knapsack.py

54 lines
1.4 KiB
Python
Raw Normal View History

2018-10-19 13:21:40 +00:00
from bisect import bisect
from itertools import accumulate
2018-10-19 13:21:40 +00:00
2019-10-05 05:14:13 +00:00
def frac_knapsack(vl, wt, w, n):
"""
>>> frac_knapsack([60, 100, 120], [10, 20, 30], 50, 3)
240.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 10, 4)
105.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 8, 4)
95.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6], 8, 4)
60.0
>>> frac_knapsack([10, 40, 30], [5, 4, 6, 3], 8, 4)
60.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 0, 4)
0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 8, 0)
95.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], -8, 4)
0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 8, -4)
95.0
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 800, 4)
130
>>> frac_knapsack([10, 40, 30, 50], [5, 4, 6, 3], 8, 400)
95.0
>>> frac_knapsack("ABCD", [5, 4, 6, 3], 8, 400)
Traceback (most recent call last):
...
TypeError: unsupported operand type(s) for /: 'str' and 'int'
"""
2018-10-19 13:21:40 +00:00
r = sorted(zip(vl, wt), key=lambda x: x[0] / x[1], reverse=True)
2019-10-05 05:14:13 +00:00
vl, wt = [i[0] for i in r], [i[1] for i in r]
acc = list(accumulate(wt))
k = bisect(acc, w)
2019-10-05 05:14:13 +00:00
return (
0
if k == 0
2024-10-14 13:30:30 +00:00
else (
sum(vl[:k]) + (w - acc[k - 1]) * (vl[k]) / (wt[k])
if k != n
else sum(vl[:k])
)
2019-10-05 05:14:13 +00:00
)
if __name__ == "__main__":
import doctest
2018-10-19 13:21:40 +00:00
doctest.testmod()