JavaScript如何实现单数组中任意N个元素的所有组合生成功能

术语确认 #

你需要的功能不是笛卡尔积，正确术语是n选k组合（k-combination），指从n个元素的集合中选出k个不重复、不考虑排列顺序的元素的所有情况。
而笛卡尔积是多个集合各取1个元素组成的所有有序元组，如果你对当前数组求N次笛卡尔积，会出现大量重复元素、顺序不同的重复组合，和你的需求不符。

参数化改造代码 #

基于你现有的flatMap实现思路扩展为递归版本，支持自定义组合元素数量：

function getCombinations(arr, k) {
  // 边界情况处理
  if (k === 0) return [[]];
  if (k > arr.length) return [];
  if (k === arr.length) return [arr];

  return arr.flatMap((current, index) => {
    // 仅取当前元素之后的子数组，避免生成顺序不同的重复组合
    const restArr = arr.slice(index + 1);
    // 从剩余子数组中选k-1个元素，和当前元素拼接成完整组合
    return getCombinations(restArr, k - 1).map(item => [current, ...item]);
  });
}

使用示例 #

const array = ['one', 'two', 'three', 'four', 'five'];

// 2元素组合，和你原有代码输出完全一致
console.log(getCombinations(array, 2));
// 3元素组合
console.log(getCombinations(array, 3));
// 4元素组合
console.log(getCombinations(array, 4));

3元素组合的输出示例：

[
  ["one","two","three"],
  ["one","two","four"],
  ["one","two","five"],
  ["one","three","four"],
  ["one","three","five"],
  ["one","four","five"],
  ["two","three","four"],
  ["two","three","five"],
  ["two","four","five"],
  ["three","four","five"]
]

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