python小试牛刀： K-best算法

最新推荐文章于 2025-09-28 18:29:10 发布

原创

最新推荐文章于 2025-09-28 18:29:10 发布 · 3.5k 阅读

1 ·

CC 4.0 BY-SA版权

介绍了一种名为k-best的算法，该算法用于从多个选项中选择单位价值最大化的k个组合。通过不断调整阈值并使用类似冒泡排序的方法，算法能在较短的时间内找到最优解。适用于物品选择等问题。

"""
k-best解决的问题：
n个珠宝价值vi和重量wi；求保留k个珠宝的单位价值最大化
n,vi,wi 值域。。。
"""

def k_best(v_ls,w_ls,k):
    l,r = 0.0, 1e5 #maximun 10,0000

    while abs(r-l) > 1e-6:
        mid = (l + r)/2.0
        vw_ls = [v - mid * w for v,w in zip(v_ls, w_ls)]
        choose =