随机从数组中获取更大数字的算法

Question

我进行了研究，发现了一般用于小列表的算法。我有一些数组，例如：

arr = [1,2,3,4 .... , 96,97,98,99,100];

arr2 = [105, 110, 165, 170];

arr3 = [1,2,7,8,9];

我想将这些数组发送到一个函数并从这个数组中获取随机数，但我希望每次都有更高的概率获得更大的数字。

比如数组1，96的概率应该大于4，但是97的概率应该大于96。

How to generate a random weighted distribution of elements

通常解决方案类似于本主题。但是，这可能会导致我的数组出现性能问题。

我怎样才能做到这一点？

Answer 1

您可以计算概率并选择值。

要将其与自定义数组一起使用，请使用 random

getRandomValue = random(sortedArray),

然后在循环中

let value = getRandomValue();

const
    random = sortedValues => {
        let length = sortedValues.length,
            sum = length * (length + 1) / 2;

        return () => {
            let r = Math.random();
            return sortedValues.find((_, i) => (r -= (i + 1) / sum) <= 0);
        };
    },
    getRandomValue = random([6, 7, 8, 9, 10]),
    counts = {};

for (let i = 0; i < 1e6; i++) {
    let value = getRandomValue();
    counts[value] = (counts[value] || 0) + 1;
}

console.log(counts);

.as-console-wrapper { max-height: 100% !important; top: 0; }