获取数组的所有（数量）组合

Question

从昨天开始我就一直在尝试完成这个，虽然还没有运气。我找到了解决方案，但我想要完成的任务总是略有不同。

我试图获得所有可能的组合，有点像这样：combination_k，但我也希望相同的项目与自身配对，所以给出以下内容：

输入[1, 4, 5] 和2（组合数）应返回：

[1, 1], [1, 4], [1, 5], [4, 4], [4, 5], [5, 5]

输入 [1, 4, 5] 和 3 应该返回：

[1, 1, 1], [1, 1, 4], [1, 1, 5], [1, 4, 4], [1, 4, 5], [4, 4, 4], [4, 4, 5], [5, 5, 5], [5, 5, 4], [5, 5, 1]（顺序不重要）。

我一直在调整combination_k，它让我足够远，它可以与2一起使用，但是当我提供3作为参数时它就不起作用了。

const combinations = getAllCombinations([1, 4, 5], 2);
// combinations = [1, 1], [1, 4], [1, 5], [4, 4], [4, 5], [5, 5]

欢迎任何提示！

Answer 1

这个问题通常被称为k-combinations with repeats。

这是一个依赖递归来获得所需结果的解决方案：

const combinations = (array, r) => {
  const result = [];
  const fn = (array, selected, c, r, start, end) => {
    if (c == r) {
      result.push([...selected]);
      return;
    }
    
    for (let i = start; i <= end; i++) {
      selected[c] = array[i];
      fn(array, selected, c + 1, r, i, end);
    }
  }
  
  fn(array, [], 0, r, 0, array.length - 1);
  return result;
}

console.log(combinations([1, 4, 5], 3));

Answer 2

您提供的代码的修改版本：

function getAllCombinations(arr, n) {
  if (n <= 0) return [];
  if (n === 1) return [...arr];

  return arr.reduce((acc, cur, i) => {
    const head = arr.slice(i, i + 1);
    const combinations = getAllCombinations(arr.slice(i), n - 1)
      .map(x => head.concat(x));
    return [...acc, ...combinations];
  }, []);
}

console.log(getAllCombinations([1, 4, 5], 2).join('|'));
console.log(getAllCombinations([1, 4, 5], 3).join('|'));