javascript - 获取给定概率的伪随机项

Question

我想在用户登录的时候给他一个奖品；但它需要有一些稀有奖品所以我想使用百分比出现不同机会的奖品

我想显示其中一个[50 : '花'], [30 : '书'], [20 : '手机'];使用他们拥有的百分比

如果有任何方法可以使用 Node.js 或仅使用 javascript 函数，那就太好了

Answer 1

您可以创建一个函数来获取加权随机结果，如下所示:

const weightedSample = (items) => {
    // cache if necessary; in Chrome, seems to make little difference
    const total = Object.values(items).reduce((sum, weight) => sum + weight, 0)

    const rnd = Math.random() * total
    let accumulator = 0

    for (const [item, weight] of Object.entries(items)) {
        accumulator += weight

        if (rnd < accumulator) {
            return item
        }
    }
}

// check frequencies of each result

const prizes = { flower: 50, book: 30, mobile: 20 }
const results = Object.fromEntries(Object.keys(prizes).map(k => [k, 0]))

for (let i = 0; i < 1e6; ++i) {
    const prize = weightedSample(prizes)
    
    ++results[prize]
}

// sample results: { flower: 500287, book: 299478, mobile: 200235 }
console.log(results)

无论权重加起来是否为 100、是否为整数等，这都有效。