JavaScript中如何实现基于加权概率的列表/数组随机器

实现思路 #

• 核心逻辑优先满足首位元素的加权概率要求，剩余元素可按需选择随机打乱或保留原有相对顺序
• 采用轮盘赌算法实现加权抽样，无需提前手动将权重归一化到总和为1，适配性更高
• 剩余元素打乱使用无偏的Fisher-Yates算法，避免常规排序打乱带来的概率偏差

完整代码实现 #

/**
 * 加权随机排序数组，首位按指定权重概率生成
 * @param {Array} arr 原始数组
 * @param {Array<number>} weights 每个元素对应出现在首位的权重，顺序和原数组一一对应
 * @param {boolean} shuffleRest 剩余元素是否随机打乱，默认true
 * @returns {Array} 排序后的数组
 */
function weightedShuffle(arr, weights, shuffleRest = true) {
  if (arr.length !== weights.length) {
    throw new Error('权重数组长度必须和原数组长度一致');
  }

  // 计算权重总和
  const totalWeight = weights.reduce((sum, w) => sum + w, 0);
  let random = Math.random() * totalWeight;
  let selectedIndex = 0;

  // 轮盘赌匹配选中的首位元素索引
  for (let i = 0; i < weights.length; i++) {
    random -= weights[i];
    if (random <= 0) {
      selectedIndex = i;
      break;
    }
  }

  // 拼接结果数组
  const result = [arr[selectedIndex]];
  const rest = arr.filter((_, index) => index !== selectedIndex);

  // 剩余元素按需打乱
  if (shuffleRest) {
    // Fisher-Yates 无偏打乱算法
    for (let i = rest.length - 1; i > 0; i--) {
      const j = Math.floor(Math.random() * (i + 1));
      [rest[i], rest[j]] = [rest[j], rest[i]];
    }
  }

  return result.concat(rest);
}

使用示例 #

对应你提到的5个元素场景：1个元素25%概率排在首位，剩余4个各18.75%概率排在首位，调用方式如下：

const list = ['元素1', '元素2', '元素3', '元素4', '元素5'];
// 权重可以直接写概率值，也可以用整数比例，比如[4,3,3,3,3]效果完全一致
const weights = [0.25, 0.1875, 0.1875, 0.1875, 0.1875];

const res = weightedShuffle(list, weights);
console.log(res);

可选调整说明 #

• 如果不需要剩余元素打乱，调用时传入第三个参数false即可，剩余元素会保留原数组的相对顺序
• 权重支持任意正数，不需要手动总和为1，比如三个元素权重设为[1,2,3]，对应首位概率就是1/6、2/6、3/6

内容的提问来源于stack exchange，提问作者sgm2000