如何实现动态嵌套数组各子数组元素的所有组合取值？

你要实现的是多个数组的笛卡尔积计算，以下是适配任意长度嵌套数组的JavaScript实现：

核心思路 #

• 从第一个子数组开始逐步拼接后续子数组的元素，每次迭代都生成当前已处理子数组的所有组合
• 不需要提前预知输入数组长度、子数组长度，完全适配动态输入场景

实现代码 #

function getFullCombinations(arrays) {
  // 输入为空直接返回空结果
  if (!arrays || arrays.length === 0) return []
  // 初始化结果为第一个子数组的单个元素组合
  let result = arrays[0].map(item => [item])
  // 遍历后续所有子数组做组合拼接
  for (let i = 1; i < arrays.length; i++) {
    const currentSubArr = arrays[i]
    const tempResult = []
    // 已有每个组合都和当前子数组的每个元素拼接生成新组合
    result.forEach(existComb => {
      currentSubArr.forEach(currentItem => {
        tempResult.push([...existComb, currentItem])
      })
    })
    result = tempResult
  }
  return result
}

使用示例 #

const myArray = [[1, 2, 3, 5], [3, 4, 5], [2, 5]]
console.log(getFullCombinations(myArray))
/* 输出符合你需要的所有组合：
[
  [1,3,2], [1,3,5], [1,4,2], [1,4,5], [1,5,2], [1,5,5],
  [2,3,2], [2,3,5], [2,4,2], [2,4,5], [2,5,2], [2,5,5],
  [3,3,2], [3,3,5], [3,4,2], [3,4,5], [3,5,2], [3,5,5],
  [5,3,2], [5,3,5], [5,4,2], [5,4,5], [5,5,2], [5,5,5]
]
*/

补充说明 #

• 如果输入存在空的子数组，默认会返回空结果，你可以根据业务需求添加过滤空数组的前置逻辑
• 迭代实现比递归实现内存占用更低，适合处理子数组元素较多的场景

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