以函数方式获取子集

Question

我需要实现一个函数，该函数获取一个由n个元素和一个数字k组成的数组A作为输入，并返回一个包含大小为k的所有子集的数组（每个子集本身就是一个数组）。

定义函数的类型，并为函数实现至少3个测试（使用assert）。它应该是Javascript/Typescript和functional

例如：FunSubset（[1,2,3]，2）=

有什么想法吗？

Answer 1

在数组中循环，并为每个元素再次循环，以创建一个长度为k-1的切片，该切片与第一个循环的当前元素连接，然后将其推送到最后返回的临时数组中。

let myArr = [1, 2, 3, 4];

let k = 2;

function subSet(arr, k) {

  let tmpArr = [];
  for (var i = 0; i < arr.length; i++) {
    for (let j = i + 1; j < arr.length; j++) {
      let slice = arr.slice(j, j + k - 1);
      slice.push(arr[i]);
      tmpArr.push(slice.reverse());
    }
  }
  return tmpArr;
}

let mySubSets = subSet(myArr, k);

console.log(JSON.stringify(mySubSets));

Answer 2

您可以为临时结果集使用带有另一个参数的生成器函数。

function* subsets(values, size, parts = []) {
    var i;

    size--;
    for (i = 0; i < values.length; i++) {
        if (size) {
            yield* subsets(values.slice(i + 1), size, [...parts, values[i]]);
        } else {
            yield [...parts, values[i]];
        }
    }
}

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

css prettyprint-override">.as-console-wrapper { max-height: 100% !important; top: 0; }