python - 以相同的方式、算法或数学方程式从列表中获取元素

Question

我有一个元素列表，例如:[1, 10, 10]，它表示我拥有的每种类型的产品数量。用户想要购买每种类型的 10 产品，均等。因此结果列表将是 [1, 5, 4] 或 [1, 4, 5]。我用了一段时间和 for 循环，迭代列表中的所有元素，如果我没有达到最大数量，则增加计数器，直到我得到 10 个产品，或者所有库存都为空。

但是我在非常大的数据的情况下有一些性能问题，我需要在多个位置应用它。

我的问题是，我是否可以用数学方程式或更好的算法来做到这一点。对于数学方程式，我提取了要求解的数据，但未知变量太多:

21/x = a
21/y = b
21/z = c
a + b + c = 10

其中21是所有产品的总和，10是用户需要的产品数量。

Answer 1

O(n) 解决方案(具有 O(n log(n)) 排序):

amounts = [1, 10, 10]
left_to_buy = 12

sorted_amounts = list(sorted([amount, index] for index, amount in enumerate(amounts)))

for idx, (amount, _) in enumerate(sorted_amounts):
    buy = min(left_to_buy//(len(amounts)-idx), amount)
    sorted_amounts[idx][0] -= buy
    left_to_buy -= buy

final_amounts = [a for a, i in sorted(sorted_amounts, key = lambda t: t[1])]
bought = [amount - final_amounts[i] for i, amount in enumerate(amounts)]

print(final_amounts, bought)
#[0, 5, 4] [1, 5, 6]

我们首先对可用项目列表进行排序(我们将它们的索引保存在一个元组中，以便最后将列表放回原始顺序)。

然后，我们从每件商品中移除它在剩余可购买商品中的份额，或者如果我们没有足够的则全部移除。

最后，我们把列表放回原来的顺序

编辑:我错过了你的列表已经排序的事实......在这种情况下，它变得更短:

amounts = [1, 10, 10]
left_to_buy = 12

res = [] 

for idx, amount in enumerate(amounts): 
    buy = min(left_to_buy//(len(amounts)-idx), amount) 
    left_to_buy -= buy 
    res.append(buy) 

print(res) 
# [1, 5, 6]