一、介绍
加权随机函数是一种在给定一组选项及其对应权重的情况下,按照权重随机选择一个选项的方法。这在很多情况下都很有用,比如在游戏中生成随机事件、在广告推荐系统中根据用户偏好选择广告、随机动画效果实现等。
二、实现
- 有一个权重数组,weight
- 有一个选项数组,items
- 随机获取一个选项
思路:
计算得到权重之和 totalWeight,然后取 [0, totalWeight) 之间随机值,根据随机值所处区间,获取对应的选项。
示例:
const weights = [1, 2, 3]; // 权重数组
const items = ['A', 'B', 'C']; // 对应的选项数组
totalWeight = 6
random = 5.447753903951361
(weights[1]=2 < random < weights[2]=3)
因此取 items[1] = 'B'
具体实现:
// 示例
const weights = [1, 2, 3]; // 权重数组
const items = ['A', 'B', 'C']; // 对应的选项数组
// 生成加权随机结果
const result = weightedRandom(weights, items);
console.log(result); // 输出随机选择的选项
function weightedRandom(weights, items) {
if (weights.length !== items.length) {
throw new Error('weights and items arrays must have the same lenght.');
}
// 计算权重总和
const totalWeight = weights.reduce((acc, val) => acc + val,0);
// 生成一个 0 到 totalWeight 之间的随机数
const random = Math.random() * totalWeight;
let weightSum = 0;
for (let i = 0; i < weights.length; i++) {
weightSum += weights[i]; // [0, 1), [1, 3), [3, 6)...
if (random < weightSum) {
return items[i];
}
}
}