Декартово произведение нескольких массивов в JavaScript

112

Как бы вы реализовали декартово произведение нескольких массивов в JavaScript?

Например,

cartesian([1, 2], [10, 20], [100, 200, 300]) 

должен вернуться

[
  [1, 10, 100],
  [1, 10, 200],
  [1, 10, 300],
  [2, 10, 100],
  [2, 10, 200]
  ...
]
Viebel
источник
возможный дубликат Найти все комбинации вариантов в цикле
Берги
3
Это реализовано в модуле js-combinatorics: github.com/dankogai/js-combinatorics
Erel Segal-Halevi
Я согласен с underscore.js, но не уверен, что понимаю, как удаление тега функционального программирования поможет @le_m
viebel
Fwiw, d3 добавлен d3.cross(a, b[, reducer])в феврале. github.com/d3/d3-array#cross
Тоф

Ответы:

106

Обновление 2017: 2-строчный ответ с vanilla JS

Все ответы здесь слишком сложны , большинство из них занимает 20 строк кода или даже больше.

В этом примере используются всего две строки ванильного JavaScript , без lodash, подчеркивания или других библиотек:

let f = (a, b) => [].concat(...a.map(a => b.map(b => [].concat(a, b))));
let cartesian = (a, b, ...c) => b ? cartesian(f(a, b), ...c) : a;

Обновить:

Это то же самое, что и выше, но улучшено, чтобы строго следовать Руководству по стилю JavaScript Airbnb - проверено с помощью ESLint с eslint-config-airbnb-base :

const f = (a, b) => [].concat(...a.map(d => b.map(e => [].concat(d, e))));
const cartesian = (a, b, ...c) => (b ? cartesian(f(a, b), ...c) : a);

Особая благодарность ZuBB за сообщение о проблемах линтера с исходным кодом.

пример

Это точный пример из вашего вопроса:

let output = cartesian([1,2],[10,20],[100,200,300]);

Вывод

Это результат этой команды:

[ [ 1, 10, 100 ],
  [ 1, 10, 200 ],
  [ 1, 10, 300 ],
  [ 1, 20, 100 ],
  [ 1, 20, 200 ],
  [ 1, 20, 300 ],
  [ 2, 10, 100 ],
  [ 2, 10, 200 ],
  [ 2, 10, 300 ],
  [ 2, 20, 100 ],
  [ 2, 20, 200 ],
  [ 2, 20, 300 ] ]

Демо

Смотрите демонстрации на:

Синтаксис

В синтаксисе, который я использовал здесь, нет ничего нового. В моем примере используется оператор распространения и остальные параметры - функции JavaScript, определенные в 6-м издании стандарта ECMA-262, опубликованного в июне 2015 года и разработанного намного раньше, более известного как ES6 или ES2015. Видеть:

Он делает такой код настолько простым, что грех не использовать его. Для старых платформ, которые не поддерживают его изначально, вы всегда можете использовать Babel или другие инструменты для преобразования его в старый синтаксис - и на самом деле мой пример, перенесенный Babel, все еще короче и проще, чем большинство примеров здесь, но это не так. действительно имеет значение, потому что результат транспиляции - это не то, что вам нужно понимать или поддерживать, это просто факт, который я нашел интересным.

Вывод

Нет необходимости писать сотни строк кода, который трудно поддерживать, и нет необходимости использовать целые библиотеки для такой простой вещи, когда две строки ванильного JavaScript могут легко выполнить работу. Как видите, использование современных функций языка действительно окупается, и в случаях, когда вам нужно поддерживать архаичные платформы без встроенной поддержки современных функций, вы всегда можете использовать Babel или другие инструменты для переноса нового синтаксиса в старый. .

Не пиши код, как будто это 1995 год

JavaScript развивается, и на это есть причина. TC39 отлично справляется с дизайном языка, добавляя новые функции, а поставщики браузеров проделывают потрясающую работу по реализации этих функций.

Чтобы увидеть текущее состояние встроенной поддержки любой данной функции в браузерах, см.

Чтобы увидеть поддержку в версиях Node, см .:

Чтобы использовать современный синтаксис на платформах, которые не поддерживают его изначально, используйте Babel:

rsp
источник
Вот версия машинописного текста с небольшими изменениями, чтобы учесть способ распространения массива в машинописном тексте. gist.github.com/ssippe/1f92625532eef28be6974f898efb23ef
Сэм
1
@rsp большое спасибо за действительно хороший ответ. хотя я хотел бы попросить вас немного улучшить его, чтобы получить предупреждение о затененных переменных (2 локальные вары aи 2 локальные вары b)
ZuBB
7
«Не пишите код так, как будто это 1995 год» - не надо показаться неприятным, еще не все догнали.
Godwhacker 01
7
Это нормально, но не помогает при кормлении, ['a', 'b'], [1,2], [[9], [10]]что [ [ 'a', 1, 9 ], [ 'a', 1, 10 ], [ 'a', 2, 9 ], [ 'a', 2, 10 ], [ 'b', 1, 9 ], [ 'b', 1, 10 ], [ 'b', 2, 9 ], [ 'b', 2, 10 ] ]в результате дает урожай . Я имею в виду, что не буду хранить предметы типа [[9], [10]].
Реду
1
Поскольку мы ...уже пользуемся , не должно ли [].concat(...[array])становиться просто [...array]?
Lazar Ljubenović
89

Вот функциональное решение проблемы (без какой-либо изменяемой переменной !) С использованием reduceи flatten, предоставляемое underscore.js:

function cartesianProductOf() {
    return _.reduce(arguments, function(a, b) {
        return _.flatten(_.map(a, function(x) {
            return _.map(b, function(y) {
                return x.concat([y]);
            });
        }), true);
    }, [ [] ]);
}

// [[1,3,"a"],[1,3,"b"],[1,4,"a"],[1,4,"b"],[2,3,"a"],[2,3,"b"],[2,4,"a"],[2,4,"b"]]
console.log(cartesianProductOf([1, 2], [3, 4], ['a']));  
<script src="https://cdnjs.cloudflare.com/ajax/libs/underscore.js/1.9.1/underscore.js"></script>

Примечание: это решение было вдохновлено http://cwestblog.com/2011/05/02/cartesian-product-of-multiple-arrays/

Viebel
источник
В этом ответе есть опечатка, не должно быть «, правда» (может быть, lodash изменился с тех пор, как вы разместили этот пост?)
Крис Джефферсон
@ChrisJefferson, второй параметр flatten- сделать сглаживание неглубоким. Здесь это обязательно!
viebel
4
Извините, это несовместимость lodash / подчеркивания, они поменяли местами флаг.
Крис Джефферсон
1
Поэтому при сведении используйте trueс подчеркиванием и используйте falseс lodash, чтобы обеспечить неглубокое сглаживание.
Аксели Пален
Как изменить эту функцию, чтобы она принимала массив массивов?
44

Вот модифицированная версия кода @ viebel на простом Javascript, без использования какой-либо библиотеки:

function cartesianProduct(arr) {
    return arr.reduce(function(a,b){
        return a.map(function(x){
            return b.map(function(y){
                return x.concat([y]);
            })
        }).reduce(function(a,b){ return a.concat(b) },[])
    }, [[]])
}

var a = cartesianProduct([[1, 2,3], [4, 5,6], [7, 8], [9,10]]);
console.log(JSON.stringify(a));

Дэнни
источник
2
Не выполняется для cartesianProduct ([[[1], [2], [3]], ['a', 'b'], [['gamma'], [['alpha']]]], ['zii', 'faa']]), поскольку он сглаживает ['gamma'] до 'gamma' и [['alpha']] до ['alpha']
Mzn
потому что .concat(y)вместо.concat([ y ])
Спасибо
@ Спасибо, вы можете отредактировать ответ напрямую, вместо того, чтобы комментировать, просто сделал это, теперь нет необходимости: P
Оливье Лалонд
28

кажется, сообщество считает это тривиальным и / или легко найти эталонную реализацию, после краткого осмотра я не смог или, может быть, просто мне нравится заново изобретать колесо или решать задачи программирования, подобные классным, в любом случае это ваш счастливый день :

function cartProd(paramArray) {

  function addTo(curr, args) {

    var i, copy, 
        rest = args.slice(1),
        last = !rest.length,
        result = [];

    for (i = 0; i < args[0].length; i++) {

      copy = curr.slice();
      copy.push(args[0][i]);

      if (last) {
        result.push(copy);

      } else {
        result = result.concat(addTo(copy, rest));
      }
    }

    return result;
  }


  return addTo([], Array.prototype.slice.call(arguments));
}


>> console.log(cartProd([1,2], [10,20], [100,200,300]));
>> [
     [1, 10, 100], [1, 10, 200], [1, 10, 300], [1, 20, 100], 
     [1, 20, 200], [1, 20, 300], [2, 10, 100], [2, 10, 200], 
     [2, 10, 300], [2, 20, 100], [2, 20, 200], [2, 20, 300]
   ]

полная эталонная реализация, которая относительно эффективна ... :-D

по эффективности: вы могли бы получить некоторые, вынув if из цикла и имея 2 отдельных цикла, поскольку он технически постоянен, и вы бы помогли с предсказанием ветвления и всем этим беспорядком, но эта точка является своего рода спором в javascript

Anywho, наслаждайтесь -ck

ckozl
источник
1
Спасибо @ckoz за подробный ответ. Почему бы вам не использовать reduceфункцию массива?
viebel 06
1
@viebel, почему вы хотите использовать сокращение? например, reduce имеет очень плохую поддержку для старых браузеров (см .: developer.mozilla.org/en-US/docs/JavaScript/Reference/… ), и в любом случае этот сумасшедший код из этого другого ответа действительно выглядит читабельным для вас ? это не для меня. Конечно, он короче, но после минимизации этот код будет примерно такой же длины, его будет легче отлаживать / оптимизировать, во-вторых, все эти решения по сокращению разбиваются на одно и то же, за исключением того, что у них есть поиск закрытия (теоретически медленнее), это также сложнее спроектировать так, чтобы он обрабатывал бесконечные множества ...
ckozl
5
Я создал в 2+ раза более быструю и (imo) более чистую версию: pastebin.com/YbhqZuf7. Он обеспечивает повышение скорости за счет того, что не использует result = result.concat(...)и не использует args.slice(1). К сожалению, мне не удалось найти способ избавиться от curr.slice()рекурсии.
Pauan 01
2
@Pauan хорошая работа, хорошее сокращение горячих точек в целом для лиги повышения производительности на 10% -50% в зависимости от того, что я вижу. Я не могу говорить о "чистоте", хотя я чувствую, что вашу версию на самом деле труднее отслеживать из-за использования переменных области видимости закрытия. Но, вообще говоря, сложнее следовать более производительному коду. Я написал исходную версию для удобства чтения, мне бы хотелось, чтобы у меня было больше времени, чтобы я мог вовлечь вас в представление;) может быть позже ...
ckozl 04
это действительно одна из тех проблем
Джеймс,
26

Следующая эффективная функция генератора возвращает декартово произведение всех заданных итераций :

// Generate cartesian product of given iterables:
function* cartesian(head, ...tail) {
  const remainder = tail.length > 0 ? cartesian(...tail) : [[]];
  for (let r of remainder) for (let h of head) yield [h, ...r];
}

// Example:
console.log(...cartesian([1, 2], [10, 20], [100, 200, 300]));

Он принимает массивы, строки, наборы и все другие объекты, реализующие итеративный протокол .

Следуя спецификации n-арного декартова произведения, получается

  • []если один или несколько заданных итераций пусты, например, []или''
  • [[a]]если дана единственная итерация, содержащая единственное значение a.

Все остальные случаи обрабатываются должным образом, как показано в следующих тестовых примерах:

le_m
источник
Не могли бы вы объяснить, что происходит на этом? Большое спасибо!
LeandroP
Спасибо за то, что научили нас замечательному примеру использования функции генератора + хвостовой рекурсии + двухслойных циклов! Но положение первого цикла for в коде необходимо изменить, чтобы упорядочить выходные подмассивы. Фиксированный код:function* cartesian(head, ...tail) { for (let h of head) { const remainder = tail.length > 0 ? cartesian(...tail) : [[]]; for (let r of remainder) yield [h, ...r] } }
ooo
@ooo Если вы хотите воспроизвести порядок кортежей декартовых произведений, заданный комментарием OP, то ваша модификация верна. Однако порядок кортежей в продукте обычно не имеет значения, например, математически результатом является неупорядоченный набор. Я выбрал этот порядок, потому что он требует гораздо меньше рекурсивных вызовов и, следовательно, немного более производительный - хотя я не запускал тест.
le_m
Ошибка: в моем комментарии выше «хвостовая рекурсия» должна быть «рекурсией» (в данном случае не хвостовым вызовом).
ooo
21

Вот простое рекурсивное решение:

function cartesianProduct(a) { // a = array of array
    var i, j, l, m, a1, o = [];
    if (!a || a.length == 0) return a;

    a1 = a.splice(0, 1)[0]; // the first array of a
    a = cartesianProduct(a);
    for (i = 0, l = a1.length; i < l; i++) {
        if (a && a.length) for (j = 0, m = a.length; j < m; j++)
            o.push([a1[i]].concat(a[j]));
        else
            o.push([a1[i]]);
    }
    return o;
}

console.log(cartesianProduct([[1,2], [10,20], [100,200,300]]));
// [[1,10,100],[1,10,200],[1,10,300],[1,20,100],[1,20,200],[1,20,300],[2,10,100],[2,10,200],[2,10,300],[2,20,100],[2,20,200],[2,20,300]]

себнукем
источник
2
Это самый эффективный чистый JS-код в этой теме. Требуется около 600 мсек для обработки массивов элементов 3 x 100 для создания массива длиной 1M.
Реду
1
Работает для cartesianProduct ([[[1], [2], [3]], ['a', 'b'], [['gamma'], [['alpha']]], ['zii', 'фаа']]); без сглаживания исходных значений
Mzn
10

Вот рекурсивный способ, в котором используется функция генератора ECMAScript 2015, поэтому вам не нужно создавать все кортежи сразу:

function* cartesian() {
    let arrays = arguments;
    function* doCartesian(i, prod) {
        if (i == arrays.length) {
            yield prod;
        } else {
            for (let j = 0; j < arrays[i].length; j++) {
                yield* doCartesian(i + 1, prod.concat([arrays[i][j]]));
            }
        }
    }
    yield* doCartesian(0, []);
}

console.log(JSON.stringify(Array.from(cartesian([1,2],[10,20],[100,200,300]))));
console.log(JSON.stringify(Array.from(cartesian([[1],[2]],[10,20],[100,200,300]))));

Heenenee
источник
Это не сработает, если в одном из массивов есть такие элементы массива, какcartesian([[1],[2]],[10,20],[100,200,300])
Redu
Ответ @Redu был обновлен для поддержки аргументов массива.
heenenee 02
Да, .concat()встроенный оператор распространения иногда может оказаться обманчивым.
Реду
10

Вот однострочник с использованием собственного ES2019 flatMap. Никаких библиотек не требуется, только современный браузер (или транспилятор):

data.reduce((a, b) => a.flatMap(x => b.map(y => [...x, y])), [[]]);

По сути, это современная версия ответа viebel без lodash.

Фред Клевер
источник
Конечно, библиотека не нужна. Но это также не самый читаемый код. Это компромисс.
Артуро Эрнандес,
В данном случае удобочитаемость больше связана с моим выбором использования оператора распространения, я думаю, а не столько с выбором отказа от использования библиотеки. Я не думаю, что lodash вообще приводит к более читаемому коду.
Фред Клевер,
9

Используя типичный возврат с помощью генераторов ES6,

function cartesianProduct(...arrays) {
  let current = new Array(arrays.length);
  return (function* backtracking(index) {
    if(index == arrays.length) yield current.slice();
    else for(let num of arrays[index]) {
      current[index] = num;
      yield* backtracking(index+1);
    }
  })(0);
}
for (let item of cartesianProduct([1,2],[10,20],[100,200,300])) {
  console.log('[' + item.join(', ') + ']');
}
div.as-console-wrapper { max-height: 100%; }

Ниже представлена ​​аналогичная версия, совместимая со старыми браузерами.

Ориол
источник
9

Это чистое решение ES6 с использованием стрелочных функций

function cartesianProduct(arr) {
  return arr.reduce((a, b) =>
    a.map(x => b.map(y => x.concat(y)))
    .reduce((a, b) => a.concat(b), []), [[]]);
}

var arr = [[1, 2], [10, 20], [100, 200, 300]];
console.log(JSON.stringify(cartesianProduct(arr)));

Кристофер Мур
источник
7

Версия coffeescript с lodash:

_ = require("lodash")
cartesianProduct = ->
    return _.reduceRight(arguments, (a,b) ->
        _.flatten(_.map(a,(x) -> _.map b, (y) -> x.concat(y)), true)
    , [ [] ])
dsummersl
источник
7

Однострочный подход с отступами для лучшего чтения.

result = data.reduce(
    (a, b) => a.reduce(
        (r, v) => r.concat(b.map(w => [].concat(v, w))),
        []
    )
);

Требуется один массив с массивами требуемых декартовых элементов.

var data = [[1, 2], [10, 20], [100, 200, 300]],
    result = data.reduce((a, b) => a.reduce((r, v) => r.concat(b.map(w => [].concat(v, w))), []));

console.log(result.map(a => a.join(' ')));
.as-console-wrapper { max-height: 100% !important; top: 0; }

Нина Шольц
источник
1
Мне пришлось добавить оператор защиты, чтобы правильно обработать случай, когда массив состоит из одного элемента:if (arr.length === 1) return arr[0].map(el => [el]);
JacobEvelyn
5

Это теговое функциональное программирование, поэтому давайте взглянем на монаду List :

Одно приложение для этого монадического списка представляет недетерминированные вычисления. List может содержать результаты для всех путей выполнения в алгоритме ...

Что ж, похоже, идеально подходит для cartesian. JavaScript дает нам, Arrayа функция монадического связывания есть Array.prototype.flatMap, поэтому давайте воспользуемся ими -

const cartesian = (...all) =>
{ const loop = (t, a, ...more) =>
    a === undefined
      ? [ t ]
      : a .flatMap (x => loop ([ ...t, x ], ...more))
  return loop ([], ...all)
}

console .log (cartesian ([1,2], [10,20], [100,200,300]))

Вместо loopвышеуказанного tможно добавить как параметр каррирования -

const makeCartesian = (t = []) => (a, ...more) =>
  a === undefined
    ? [ t ]
    : a .flatMap (x => makeCartesian ([ ...t, x ]) (...more))

const cartesian =
  makeCartesian ()

console .log (cartesian ([1,2], [10,20], [100,200,300]))

Спасибо
источник
3

Некоторые ответы по этой теме не работают, если какой-либо из входных массивов содержит элемент массива. Тебе лучше это проверить.

В любом случае нет необходимости в подчеркивании, вообще в lodash. Я считаю, что этот должен делать это на чистом JS ES6, настолько функциональном, насколько это возможно.

Этот фрагмент кода использует сокращение и вложенную карту, просто чтобы получить декартово произведение двух массивов, однако второй массив получается в результате рекурсивного вызова той же функции с одним массивом меньше; следовательно .. a[0].cartesian(...a.slice(1))

Array.prototype.cartesian = function(...a){
  return a.length ? this.reduce((p,c) => (p.push(...a[0].cartesian(...a.slice(1)).map(e => a.length > 1 ? [c,...e] : [c,e])),p),[])
                  : this;
};

var arr = ['a', 'b', 'c'],
    brr = [1,2,3],
    crr = [[9],[8],[7]];
console.log(JSON.stringify(arr.cartesian(brr,crr))); 

Уменьшить
источник
3

В моем конкретном случае «старомодный» подход казался более эффективным, чем методы, основанные на более современных функциях. Ниже приведен код (включая небольшое сравнение с другими решениями, опубликованными в этой ветке @rsp и @sebnukem), если он окажется полезным и для кого-то другого.

Идея следующая. Допустим, мы создаем внешний продукт Nмассивов, a_1,...,a_Nкаждый из которых имеет m_iкомпоненты. Внешний продукт этих массивов имеет M=m_1*m_2*...*m_Nэлементы, и мы можем идентифицировать каждый из них с помощью N-размерного вектора, компоненты которого являются положительными целыми числами, а i-я компонента строго ограничена сверху значением m_i. Например, вектор (0, 0, ..., 0)будет соответствовать конкретной комбинации, в которой берется первый элемент из каждого массива, а (m_1-1, m_2-1, ..., m_N-1)идентифицируется комбинацией, в которой берется последний элемент из каждого массива. Таким образом, чтобы построить всеM комбинации, функция ниже последовательно создает все такие векторы и для каждого из них идентифицирует соответствующую комбинацию элементов входных массивов.

function cartesianProduct(){
    const N = arguments.length;

    var arr_lengths = Array(N);
    var digits = Array(N);
    var num_tot = 1;
    for(var i = 0; i < N; ++i){
        const len = arguments[i].length;
        if(!len){
            num_tot = 0;
            break;
        }
        digits[i] = 0;
        num_tot *= (arr_lengths[i] = len);
    }

    var ret = Array(num_tot);
    for(var num = 0; num < num_tot; ++num){

        var item = Array(N);
        for(var j = 0; j < N; ++j){ item[j] = arguments[j][digits[j]]; }
        ret[num] = item;

        for(var idx = 0; idx < N; ++idx){
            if(digits[idx] == arr_lengths[idx]-1){
                digits[idx] = 0;
            }else{
                digits[idx] += 1;
                break;
            }
        }
    }
    return ret;
}
//------------------------------------------------------------------------------
let _f = (a, b) => [].concat(...a.map(a => b.map(b => [].concat(a, b))));
let cartesianProduct_rsp = (a, b, ...c) => b ? cartesianProduct_rsp(_f(a, b), ...c) : a;
//------------------------------------------------------------------------------
function cartesianProduct_sebnukem(a) {
    var i, j, l, m, a1, o = [];
    if (!a || a.length == 0) return a;

    a1 = a.splice(0, 1)[0];
    a = cartesianProduct_sebnukem(a);
    for (i = 0, l = a1.length; i < l; i++) {
        if (a && a.length) for (j = 0, m = a.length; j < m; j++)
            o.push([a1[i]].concat(a[j]));
        else
            o.push([a1[i]]);
    }
    return o;
}
//------------------------------------------------------------------------------
const L = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9];
const args = [L, L, L, L, L, L];

let fns = {
    'cartesianProduct': function(args){ return cartesianProduct(...args); },
    'cartesianProduct_rsp': function(args){ return cartesianProduct_rsp(...args); },
    'cartesianProduct_sebnukem': function(args){ return cartesianProduct_sebnukem(args); }
};

Object.keys(fns).forEach(fname => {
    console.time(fname);
    const ret = fns[fname](args);
    console.timeEnd(fname);
});

с node v6.12.2, я получаю следующие тайминги:

cartesianProduct: 427.378ms
cartesianProduct_rsp: 1710.829ms
cartesianProduct_sebnukem: 593.351ms
ewcz
источник
3

Для тех, кому нужен TypeScript (переработанный ответ @Danny)

/**
 * Calculates "Cartesian Product" sets.
 * @example
 *   cartesianProduct([[1,2], [4,8], [16,32]])
 *   Returns:
 *   [
 *     [1, 4, 16],
 *     [1, 4, 32],
 *     [1, 8, 16],
 *     [1, 8, 32],
 *     [2, 4, 16],
 *     [2, 4, 32],
 *     [2, 8, 16],
 *     [2, 8, 32]
 *   ]
 * @see https://stackoverflow.com/a/36234242/1955709
 * @see https://en.wikipedia.org/wiki/Cartesian_product
 * @param arr {T[][]}
 * @returns {T[][]}
 */
function cartesianProduct<T> (arr: T[][]): T[][] {
  return arr.reduce((a, b) => {
    return a.map(x => {
      return b.map(y => {
        return x.concat(y)
      })
    }).reduce((c, d) => c.concat(d), [])
  }, [[]] as T[][])
}
artnikpro
источник
2

Просто на выбор настоящая простая реализация с использованием массива reduce:

const array1 = ["day", "month", "year", "time"];
const array2 = ["from", "to"];
const process = (one, two) => [one, two].join(" ");

const product = array1.reduce((result, one) => result.concat(array2.map(two => process(one, two))), []);
Simple.Js
источник
2

Современный JavaScript всего в нескольких строках. Никаких внешних библиотек или зависимостей, таких как Lodash.

function cartesian(...arrays) {
  return arrays.reduce((a, b) => a.flatMap(x => b.map(y => x.concat([y]))), [ [] ]);
}

console.log(
  cartesian([1, 2], [10, 20], [100, 200, 300])
    .map(arr => JSON.stringify(arr))
    .join('\n')
);

Майлз Элам
источник
2

Вы могли reduceбы 2D-массив. Используйте flatMapв массиве аккумуляторов, чтобы получить acc.length x curr.lengthколичество комбинаций в каждом цикле. [].concat(c, n)используется, потому что cэто число в первой итерации, а затем - массив.

const data = [ [1, 2], [10, 20], [100, 200, 300] ];

const output = data.reduce((acc, curr) =>
  acc.flatMap(c => curr.map(n => [].concat(c, n)))
)

console.log(JSON.stringify(output))

(Это основано на ответе Нины Шольц )

Адига
источник
1

Нерекурсивный подход, который добавляет возможность фильтровать и изменять продукты перед фактическим добавлением их в набор результатов. Обратите внимание на использование .map, а не .forEach. В некоторых браузерах .map работает быстрее.

function crossproduct(arrays,rowtest,rowaction) {
      // Calculate the number of elements needed in the result
      var result_elems = 1, row_size = arrays.length;
      arrays.map(function(array) {
            result_elems *= array.length;
      });
      var temp = new Array(result_elems), result = [];

      // Go through each array and add the appropriate element to each element of the temp
      var scale_factor = result_elems;
      arrays.map(function(array)
      {
        var set_elems = array.length;
        scale_factor /= set_elems;
        for(var i=result_elems-1;i>=0;i--) {
            temp[i] = (temp[i] ? temp[i] : []);
            var pos = i / scale_factor % set_elems;
            // deal with floating point results for indexes, this took a little experimenting
            if(pos < 1 || pos % 1 <= .5) {
                pos = Math.floor(pos);
            } else {
                pos = Math.min(array.length-1,Math.ceil(pos));
            }
            temp[i].push(array[pos]);
            if(temp[i].length===row_size) {
                var pass = (rowtest ? rowtest(temp[i]) : true);
                if(pass) {
                    if(rowaction) {
                        result.push(rowaction(temp[i]));
                    } else {
                        result.push(temp[i]);
                    }
                }
            }
        }
      });
      return result;
    }
AnyWhichWay
источник
1

Простое «разумное и визуально понятное» решение.

введите описание изображения здесь


// t = [i, length]

const moveThreadForwardAt = (t, tCursor) => {
  if (tCursor < 0)
    return true; // reached end of first array

  const newIndex = (t[tCursor][0] + 1) % t[tCursor][1];
  t[tCursor][0] = newIndex;

  if (newIndex == 0)
    return moveThreadForwardAt(t, tCursor - 1);

  return false;
}

const cartesianMult = (...args) => {
  let result = [];
  const t = Array.from(Array(args.length)).map((x, i) => [0, args[i].length]);
  let reachedEndOfFirstArray = false;

  while (false == reachedEndOfFirstArray) {
    result.push(t.map((v, i) => args[i][v[0]]));

    reachedEndOfFirstArray = moveThreadForwardAt(t, args.length - 1);
  }

  return result;
}

// cartesianMult(
//   ['a1', 'b1', 'c1'],
//   ['a2', 'b2'],
//   ['a3', 'b3', 'c3'],
//   ['a4', 'b4']
// );

console.log(cartesianMult(
  ['a1'],
  ['a2', 'b2'],
  ['a3', 'b3']
));
ноль. ноль. семь
источник
1

Простая модифицированная версия кода @ viebel на простом Javascript:

function cartesianProduct(...arrays) {
  return arrays.reduce((a, b) => {
    return [].concat(...a.map(x => {
      const next = Array.isArray(x) ? x : [x];
      return [].concat(b.map(y => next.concat(...[y])));
    }));
  });
}

const product = cartesianProduct([1, 2], [10, 20], [100, 200, 300]);

console.log(product);
/*
[ [ 1, 10, 100 ],
  [ 1, 10, 200 ],
  [ 1, 10, 300 ],
  [ 1, 20, 100 ],
  [ 1, 20, 200 ],
  [ 1, 20, 300 ],
  [ 2, 10, 100 ],
  [ 2, 10, 200 ],
  [ 2, 10, 300 ],
  [ 2, 20, 100 ],
  [ 2, 20, 200 ],
  [ 2, 20, 300 ] ];
*/
Хуан Гайтан
источник
1

Более читаемая реализация

function productOfTwo(one, two) {
  return one.flatMap(x => two.map(y => [].concat(x, y)));
}

function product(head = [], ...tail) {
  if (tail.length === 0) return head;
  return productOfTwo(head, product(...tail));
}

const test = product(
  [1, 2, 3],
  ['a', 'b']
);

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

Андрей
источник
1
f=(a,b,c)=>a.flatMap(ai=>b.flatMap(bi=>c.map(ci=>[ai,bi,ci])))

Это для 3-х массивов.
Некоторые ответы давали возможность использовать любое количество массивов.
Это может легко сжиматься или расширяться до меньшего или большего количества массивов.
Мне нужны были комбинации одного подхода с повторениями, поэтому я мог использовать:

f(a,a,a)

но использовали:

f=(a,b,c)=>a.flatMap(a1=>a.flatMap(a2=>a.map(a3=>[a1,a2,a3])))
iAmOren
источник
0

Я заметил, что никто не опубликовал решение, позволяющее передавать функцию для обработки каждой комбинации, поэтому вот мое решение:

const _ = require('lodash')

function combinations(arr, f, xArr = []) {
    return arr.length>1 
    ? _.flatMap(arr[0], x => combinations(arr.slice(1), f, xArr.concat(x)))
    : arr[0].map(x => f(...xArr.concat(x)))
}

// use case
const greetings = ["Hello", "Goodbye"]
const places = ["World", "Planet"]
const punctuationMarks = ["!", "?"]
combinations([greetings,places,punctuationMarks], (greeting, place, punctuationMark) => `${greeting} ${place}${punctuationMark}`)
  .forEach(row => console.log(row))

Вывод:

Hello World!
Hello World?
Hello Planet!
Hello Planet?
Goodbye World!
Goodbye World?
Goodbye Planet!
Goodbye Planet?
Lezorte
источник
0

Простой метод грубой силы JS, который принимает в качестве входных данных массив массивов.

var cartesian = function(arrays) {
    var product = [];
    var precals = [];
    var length = arrays.reduce(function(acc, curr) {
        return acc * curr.length
    }, 1);
    for (var i = 0; i < arrays.length; i++) {
        var array = arrays[i];
        var mod = array.length;
        var div = i > 0 ? precals[i - 1].div * precals[i - 1].mod : 1;
        precals.push({
            div: div,
            mod: mod
        });
    }
    for (var j = 0; j < length; j++) {
        var item = [];
        for (var i = 0; i < arrays.length; i++) {
            var array = arrays[i];
            var precal = precals[i];
            var k = (~~(j / precal.div)) % precal.mod;
            item.push(array[k]);
        }
        product.push(item);
    }
    return product;
};

cartesian([
    [1],
    [2, 3]
]);

cartesian([
    [1],
    [2, 3],
    [4, 5, 6]
]);
Сэмюэл Вентура
источник
0

var chars = ['A', 'B', 'C']
var nums = [1, 2, 3]

var cartesianProduct = function() {
  return _.reduce(arguments, function(a, b) {
    return _.flatten(_.map(a, function(x) {
      return _.map(b, function(y) {
        return x.concat(y);
      });
    }), true);
  }, [
    []
  ]);
};

console.log(cartesianProduct(chars, nums))
<script src="https://cdnjs.cloudflare.com/ajax/libs/underscore.js/1.8.3/underscore-min.js"></script>

Просто конвертированы @ dummersl в ответ от CoffeScript на JavaScript. Просто работает.

var chars = ['A', 'B', 'C']
var nums = [1, 2, 3]

var cartesianProduct = function() {
  return _.reduce(arguments, function(a, b) {
    return _.flatten(_.map(a, function(x) {
      return _.map(b, function(y) {
        return x.concat(y);
      });
    }), true);
  }, [[]]);
};

console.log( cartesianProduct(chars, nums) )
гунейзус
источник
0

Еще одна реализация. Не самый короткий или навороченный, но быстрый:

function cartesianProduct() {
    var arr = [].slice.call(arguments),
        intLength = arr.length,
        arrHelper = [1],
        arrToReturn = [];

    for (var i = arr.length - 1; i >= 0; i--) {
        arrHelper.unshift(arrHelper[0] * arr[i].length);
    }

    for (var i = 0, l = arrHelper[0]; i < l; i++) {
        arrToReturn.push([]);
        for (var j = 0; j < intLength; j++) {
            arrToReturn[i].push(arr[j][(i / arrHelper[j + 1] | 0) % arr[j].length]);
        }
    }

    return arrToReturn;
}
вспышка256
источник
0

Библиотеки не нужны! :)

Однако нужны стрелочные функции и, вероятно, не так эффективны. : /

const flatten = (xs) => 
    xs.flat(Infinity)

const binaryCartesianProduct = (xs, ys) =>
    xs.map((xi) => ys.map((yi) => [xi, yi])).flat()

const cartesianProduct = (...xss) =>
    xss.reduce(binaryCartesianProduct, [[]]).map(flatten)
      
console.log(cartesianProduct([1,2,3], [1,2,3], [1,2,3]))

Крис Вуга
источник
0

Для записи

Вот моя версия. Я использовал простейший итератор javascript "for ()", поэтому он совместим во всех случаях и имеет лучшую производительность.

function cartesian(arrays){
    var quant = 1, counters = [], retArr = [];

    // Counts total possibilities and build the counters Array;
    for(var i=0;i<arrays.length;i++){
        counters[i] = 0;
        quant *= arrays[i].length;
    }

    // iterate all possibilities
    for(var i=0,nRow;i<quant;i++){
        nRow = [];
        for(var j=0;j<counters.length;j++){
            if(counters[j] < arrays[j].length){
                nRow.push(arrays[j][counters[j]]);
            } else { // in case there is no such an element it restarts the current counter
                counters[j] = 0;
                nRow.push(arrays[j][counters[j]]);
            }
            counters[j]++;
        }
        retArr.push(nRow);
    }
    return retArr;
}

С уважением.

Леандроп
источник