HOW - 实现加权随机函数

一、介绍

加权随机函数是一种在给定一组选项及其对应权重的情况下，按照权重随机选择一个选项的方法。这在很多情况下都很有用，比如在游戏中生成随机事件、在广告推荐系统中根据用户偏好选择广告、随机动画效果实现等。

二、实现

1. 有一个权重数组，weight
2. 有一个选项数组，items
3. 随机获取一个选项

思路：

计算得到权重之和 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];
		}
	}
}