mirror of
https://github.com/krahets/hello-algo.git
synced 2025-11-01 11:29:51 +08:00
954c45864b
* docs: add Japanese documents (`ja/docs`) * docs: add Japanese documents (`ja/codes`) * docs: add Japanese documents * Remove pythontutor blocks in ja/ * Add an empty at the end of each markdown file. * Add the missing figures (use the English version temporarily). * Add index.md for Japanese version. * Add index.html for Japanese version. * Add missing index.assets * Fix backtracking_algorithm.md for Japanese version. * Add avatar_eltociear.jpg. Fix image links on the Japanese landing page. * Add the Japanese banner. --------- Co-authored-by: krahets <krahets@163.com>
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"""
|
|
File: fractional_knapsack.py
|
|
Created Time: 2023-07-19
|
|
Author: krahets (krahets@163.com)
|
|
"""
|
|
|
|
|
|
class Item:
|
|
"""アイテム"""
|
|
|
|
def __init__(self, w: int, v: int):
|
|
self.w = w # アイテムの重量
|
|
self.v = v # アイテムの価値
|
|
|
|
|
|
def fractional_knapsack(wgt: list[int], val: list[int], cap: int) -> int:
|
|
"""分数ナップサック：貪欲法"""
|
|
# アイテムリストを作成、2 つの属性を含む：重量、価値
|
|
items = [Item(w, v) for w, v in zip(wgt, val)]
|
|
# 単位価値 item.v / item.w で高い順にソート
|
|
items.sort(key=lambda item: item.v / item.w, reverse=True)
|
|
# 貪欲選択をループ
|
|
res = 0
|
|
for item in items:
|
|
if item.w <= cap:
|
|
# 残り容量が十分な場合、アイテム全体をナップサックに入れる
|
|
res += item.v
|
|
cap -= item.w
|
|
else:
|
|
# 残り容量が不十分な場合、アイテムの一部をナップサックに入れる
|
|
res += (item.v / item.w) * cap
|
|
# 残り容量がなくなったため、ループを中断
|
|
break
|
|
return res
|
|
|
|
|
|
"""ドライバーコード"""
|
|
if __name__ == "__main__":
|
|
wgt = [10, 20, 30, 40, 50]
|
|
val = [50, 120, 150, 210, 240]
|
|
cap = 50
|
|
n = len(wgt)
|
|
|
|
# 貪欲アルゴリズム
|
|
res = fractional_knapsack(wgt, val, cap)
|
|
print(f"ナップサック容量を超えないアイテムの最大値は {res}") |