根据索引将元素均衡分配到 bin

Question

我有 n 个索引为 0..(n-1) 的元素。我想像这样将元素分发到 m 个垃圾箱：

我想按顺序填充垃圾箱
垃圾箱的大小应介于 ⌊number_of_elements / number_of_bins⌋ 和 ⌈number_of_elements / number_of_bins⌉ 之间。较大的垃圾桶应该放在第一位。
我想根据元素的索引分配元素。我只能想出各种for循环的解决方案。应该可以只使用一个 for 循环将元素分配给一个 bin 和 mod 和 div 以及可能的 if 运算符。

示例：我有 n=7 个元素和 m=3 个分箱。结果应该是这样的：

Bin 1: 0, 1, 2 
Bin 2: 3, 4
Bin 3: 5, 6

Answer 1

这是 Python 中的概念验证示例。

# Initialize
elements = [0, 1, 2, 3, 4, 5, 6];
n = len(elements); # Number of elements
m = 3; # Number of bins
bins = [[] for x in range(m)];

# Precalculate this
elementsPerBinCeil = n / m;
elementsPerBinFloor = n / m - 1;
# This is the bin number above which we have to use elementsPerBinFloor
cutoffNum = n % m;
i = 0; # This is which bin to assign the element to
# Assign all elements to a bin
for element in elements:
    bins[i].append(element);
    # Move to next bin
    if (i < cutoffNum and len(bins[i]) > elementsPerBinCeil):
        i += 1;
    elif (i >= cutoffNum and len(bins[i]) > elementsPerBinFloor):
        i += 1;

更新：我在Pythonhere中有几个示例实现。如果您对做同一件事的不同方法感兴趣，请检查存储库的各个分支。

根据索引将元素均衡分配到 bin

Balanced distribution of elements to bins based on index

element

distribution

bin