贪心算法之分数背包
·
背景
话说一个小偷在某商店发现有n个商品,第i个商品价值vi元,重wi千克。他希望拿走的价值尽量高,但他的背包最多只能容纳W千克的东西。他应该拿走哪些商品? 分数背包:对于一个商品,小偷可以拿走其中任意一部分。( 唉,这年头,小偷也不好当了!)
思想
贪心思想:因为商品是可以被拆分来拿的,所以我们可以挑那些单位价值更贵的商品先拿,遇到整个商品拿不下的情况,我们就直拿一部分即可。
源码
def fractional_backpack(goods, W):
'''
:param goods:商品列表
:param W: 背包可以容纳的重量
:return: 拿走的最大价值
'''
# 对商品按照单位价值进行降序排序
goods.sort(key=lambda a: a[0] / a[1], reverse=True)
n = [0 for i in range(len(goods))] # 记录每个商品取走多少
for i, (price, weight) in enumerate(goods):
# 贪心算法:每次总是先拿价值最大的商品
if W > weight:
n[i] = 1 # 如果背包的容量大于商品的重量,就可以全部取走
W -= weight # 拿走weight千克
else:
n[i] = W / weight # 拿走W千克
W -= W
break
res_li = [goods[i][0] * num for i, num in enumerate(n)]
print(n)
print(goods)
print(res_li)
return sum(res_li)

同步更新于个人博客系统:贪心算法之分数背包
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)