如何实现按数组索引概率递增的JavaScript CustomRandom函数

我们需要实现一个名为CustomRandom的JavaScript函数，接收字符数组作为参数，要求返回数组中的随机字符，且字符的返回概率随数组索引递增——索引越大的元素，被选中的概率越高。

举个具体例子，当传入['b','c','a','z']时：

• 索引0的b概率最低，约10%（100次调用约出现10次）
• 索引1的c概率次之，约20%
• 索引2的a概率更高，约30%
• 索引3的z概率最高，约40%
  实际结果无需严格匹配上述数值，只要满足概率递增的规则即可。

实现思路 #

核心逻辑是给每个元素分配随索引递增的权重，再通过随机数匹配对应的权重区间来选择元素：

1. 给第i个元素（从0开始）分配权重i+1，这样索引越大权重越高
2. 计算所有元素的总权重，生成0到总权重之间的随机数
3. 遍历数组，用随机数依次减去每个元素的权重，当随机数小于等于0时，返回当前元素

代码实现 #

function CustomRandom(arr) {
  // 处理空数组或无效输入
  if (!Array.isArray(arr) || arr.length === 0) return null;
  
  // 计算总权重：1+2+...+n = n*(n+1)/2
  const totalWeight = arr.length * (arr.length + 1) / 2;
  // 生成0到totalWeight之间的随机数
  let randomValue = Math.random() * totalWeight;
  
  // 遍历找到随机数对应的元素
  for (let i = 0; i < arr.length; i++) {
    const currentWeight = i + 1;
    randomValue -= currentWeight;
    if (randomValue <= 0) {
      return arr[i];
    }
  }
  
  // 兜底逻辑（理论上不会执行到）
  return arr[arr.length - 1];
}

代码说明 #

• 边界处理：先判断输入是否为有效数组，避免报错
• 权重计算：总权重公式是等差数列求和，确保权重分配符合递增要求
• 随机匹配：通过递减随机数找到对应的元素，逻辑简单高效，时间复杂度为O(n)

测试示例 #

// 模拟1000次调用统计结果
const resultCounts = { b: 0, c: 0, a: 0, z: 0 };
for (let i = 0; i < 1000; i++) {
  const selectedChar = CustomRandom(['b','c','a','z']);
  resultCounts[selectedChar]++;
}
console.log(resultCounts);
// 输出示例（非严格）：{ b: 98, c: 205, a: 302, z: 395 }

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