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或任何标准库。

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; }