JavaScript中数组对象的笛卡尔积

Question

我有以下输入。

input = [{a:[1,2]}, {b:[3,4]}];

我想使用这个输入创建一个叉积输出，其格式如下：

output = [{a:1, b:3},{a:1, b:4},{a:2, b:3},{a:2, b:4}];

输入可能是动态的

，例如：

input = [{a:[1]}, {b:[2,4]}, {c:[3,5,6]}];

可以使用Lodash或任何标准库。

- heyayush

Answer 1

您可以采用迭代和递归的方法。

function cartesian(array) {
    function c(part, index) {
        var k = Object.keys(array[index])[0];
        array[index][k].forEach(function (a) {
            var p = Object.assign({}, part, { [k]: a });
            if (index + 1 === array.length) {
                r.push(p);
                return;
            }
            c(p, index + 1);
        });
    }

    var r = [];
    c({}, 0);
    return r;
}

console.log(cartesian([{ a: [1, 2] }, { b: [3, 4] }]));
console.log(cartesian([{ a: [1] }, { b: [2, 4] }, { c: [3, 5, 6] }]));

.as-console-wrapper { max-height: 100% !important; top: 0; }

- Nina Scholz

Answer 2

这是使用递归函数实现的方法。

function generate(data) {
  var r = [];
  
  function cartesian(data, n, c) {
    if(n == data.length) {
      return r.push(Object.assign({}, c));
    }
    var arr = data[n][Object.keys(data[n])[0]]
    for(var i = 0; i < arr.length; i++) {
      c[Object.keys(data[n])[0]] = arr[i] 
      cartesian(data, n + 1, c);
    }
  }
  
  cartesian(data, 0, {});
  return r;
}
var input1 = [{a:[1,2]}, {b:[3,4]}];
var input2 = [{a:[1]}, {b:[2,4]}, {c:[3,5,6]}];

console.log(generate(input1))
console.log(generate(input2))

- Nenad Vracar