如何实现满足特定约束的数组随机采样函数（函数式编程）

要满足你提出的两个约束并遵循函数式编程风格（避免for循环），我们可以通过reduce累积构建结果数组的方式实现，核心思路是每一步生成元素时，根据已生成的结果动态排除不符合要求的选项，再随机选取合法元素。

实现步骤 #

1. 带排除规则的随机元素选取函数：扩展基础的随机选取逻辑，支持传入需要排除的元素列表，确保只从合法候选集中随机选择。
2. 用reduce构建结果数组：通过reduce迭代累积结果，每一步根据当前已生成的数组，计算需要排除的元素（连续重复的前一个元素、同列的上一个元素），再调用上述函数选取合法元素。
3. 边界校验：提前处理无法满足约束的场景（比如输入数组长度为1且采样规模大于1时，必然违反约束）。

完整代码实现 #

const arr = [1, 2, 3, 4];

// 从数组中随机选取一个不在排除列表中的元素
const getRandomValidElement = (arr, excluded = []) => {
  const validCandidates = arr.filter(item => !excluded.includes(item));
  if (validCandidates.length === 0) {
    throw new Error("无法生成符合约束的结果：输入数组长度不足以满足采样规则");
  }
  return validCandidates[Math.floor(Math.random() * validCandidates.length)];
};

// 函数式实现的带约束随机采样函数
const randomize = (arr, size) => {
  // 边界校验：数组长度为1且size>1时，必然违反约束
  if (arr.length === 1 && size > 1) {
    throw new Error("输入数组长度为1时，无法满足size>1的采样约束");
  }
  
  const totalLength = size * size;
  // 用reduce累积构建结果数组，避免for循环
  return Array.from({ length: totalLength }).reduce((acc, _, index) => {
    const forbidden = [];
    // 约束1：排除前一个元素（避免连续重复）
    if (index > 0) {
      forbidden.push(acc[index - 1]);
    }
    // 约束2：排除同列的上一个元素（索引i与i-size的元素不相等）
    if (index >= size) {
      forbidden.push(acc[index - size]);
    }
    // 去重排除列表（避免重复添加同一个元素）
    const uniqueForbidden = [...new Set(forbidden)];
    // 选取合法元素并添加到结果中
    const nextElement = getRandomValidElement(arr, uniqueForbidden);
    return [...acc, nextElement];
  }, []);
};

// 示例调用
console.log(randomize(arr, 2));

代码说明 #

• 函数式特性：使用reduce替代传统循环进行状态累积，所有操作均为纯函数（除了Math.random的随机性，这是随机采样的固有特性），没有可变变量。
• 约束处理：
  • 约束1：每次迭代时，若不是第一个元素，则排除结果数组的最后一个元素（避免连续重复）。
  • 约束2：当当前索引大于等于size时，排除结果数组中index - size位置的元素（即同列的上一个元素）。
• 鲁棒性：添加了边界校验和合法候选集为空时的错误抛出，避免出现无限循环或无效结果。

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