JavaScript如何实现带分配概率的自定义数组洗牌功能

问题解答 #

结论 #

可以直接使用Math.random()实现该需求，不需要引入其他额外工具。

实现思路 #

你需要的是带权重的洗牌逻辑，有两种常见实现方案可以满足你的需求：

方案1：加权随机键排序

这种方案代码简洁，通过给不同类型的元素分配不同权重的随机排序键，实现B开头元素80%概率靠前、W开头20%概率靠前的效果，符合加权随机采样的标准逻辑：

function weightedShuffle(array) {
  // 为每个元素生成加权随机键
  const withWeight = array.map(item => {
    // B开头元素权重0.8，W开头权重0.2
    const weight = item.startsWith('B') ? 0.8 : 0.2;
    // 权重越高，生成的随机键值越大的概率越高
    const sortKey = Math.pow(Math.random(), 1 / weight);
    return { item, sortKey };
  });
  // 按键值从大到小排序，键值越大排位越靠前
  withWeight.sort((a, b) => b.sortKey - a.sortKey);
  // 提取排序后的元素数组
  return withWeight.map(entry => entry.item);
}

方案2：按位置概率分组采样

如果你希望每个靠前位置的元素类型严格贴合80%B、20%W的概率，可以用分组独立洗牌+按概率取元素的方案，你原来的普通洗牌函数可以直接复用：

// 复用你原有的普通洗牌逻辑
function shuffle(array){
  var counter = array.length,
      temp, index;
  while (counter > 0){
    index = Math.floor(Math.random() * counter);
    counter = counter-1;
    temp = array[counter];
    array[counter] = array[index];
    array[index] = temp;
  }
  return array;
}

function weightedShuffle(array) {
  // 拆分两类元素并分别做普通洗牌
  const bList = shuffle(array.filter(item => item.startsWith('B')));
  const wList = shuffle(array.filter(item => item.startsWith('W')));
  const result = [];
  // 逐个位置按概率选择元素类型
  while (bList.length || wList.length) {
    const random = Math.random();
    if (random < 0.8 && bList.length) {
      result.push(bList.pop());
    } else if (wList.length) {
      result.push(wList.pop());
    } else {
      result.push(bList.pop());
    }
  }
  return result;
}

使用方式 #

直接调用新的加权洗牌函数即可：

const shuffledArray = weightedShuffle(myArray);

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