Вывести каждую комбинацию из массива чисел с помощью JavaScript - PullRequest
6 голосов
/ 31 октября 2010

У меня есть несколько чисел в массиве

var numArr = [1, 3, 5, 9];

Я хочу перебрать этот массив и умножить каждую уникальную комбинацию из 3 чисел следующим образом:

1 * 3 * 5 = 
1 * 3 * 9 = 
1 * 5 * 9 = 
3 * 5 * 9 =

Затем вернуть массиввсе расчеты

var ansArr = [15,27,45,135];

У кого-нибудь есть элегантное решение?Заранее спасибо.

Ответы [ 6 ]

12 голосов
/ 31 октября 2010

Универсальный алгоритм генерации комбинаций выглядит следующим образом:

function combinations(numArr, choose, callback) {
    var n = numArr.length;
    var c = [];
    var inner = function(start, choose_) {
        if (choose_ == 0) {
            callback(c);
        } else {
            for (var i = start; i <= n - choose_; ++i) {
                c.push(numArr[i]);
                inner(i + 1, choose_ - 1);
                c.pop();
            }
        }
    }
    inner(0, choose);
}

В вашем случае вы можете назвать его так:

function product(arr) {
    p = 1;
    for (var i in arr) {
        p *= arr[i];
    }
    return p;
}

var ansArr = [];

combinations(
    [1, 3, 5, 7, 9, 11], 3,
    function output(arr) {
        ansArr.push(product(arr));
    });

document.write(ansArr);

... который, длязаданный вход, дает это:

15,21,27,33,35,45,55,63,77,99,105,135,165,189,231,297,315,385,495,693
2 голосов
/ 31 октября 2010

Я думаю, что это должно работать:

var a = [1, 3, 5, 9];
var l = a.length;
var r = [];
for (var i = 0; i < l; ++i) {
  for (var j = i + 1; j < l; ++j) {
    for (var k = j + 1; k < l; ++k) {
      r.push(a[i] * a[j] * a[k]);
    }
  }
}

Редактировать

Только для собственного наглядности я нашел общее решение, в котором вместо рекурсии используются циклы. Очевидным недостатком является то, что он длиннее, поэтому медленнее загружать или читать. С другой стороны (по крайней мере, на Firefox на моей машине) он работает примерно в два раза быстрее, чем рекурсивная версия. Тем не менее, я бы порекомендовал его, только если вы находите комбинации для больших наборов или находите комбинации много раз на одной странице. В любом случае, если кому-то интересно, вот что я придумала.

function combos(superset, size) {
  var result = [];
  if (superset.length < size) {return result;}
  var done = false;
  var current_combo, distance_back, new_last_index;
  var indexes = [];
  var indexes_last = size - 1;
  var superset_last = superset.length - 1;

  // initialize indexes to start with leftmost combo
  for (var i = 0; i < size; ++i) {
    indexes[i] = i;
  }

  while (!done) {
    current_combo = [];
    for (i = 0; i < size; ++i) {
      current_combo.push(superset[indexes[i]]);
    }
    result.push(current_combo);
    if (indexes[indexes_last] == superset_last) {
      done = true;
      for (i = indexes_last - 1; i > -1 ; --i) {
        distance_back = indexes_last - i;
        new_last_index = indexes[indexes_last - distance_back] + distance_back + 1;
        if (new_last_index <= superset_last) {
          indexes[indexes_last] = new_last_index;
          done = false;
          break;
        }
      }
      if (!done) {
        ++indexes[indexes_last - distance_back];
        --distance_back;
        for (; distance_back; --distance_back) {
          indexes[indexes_last - distance_back] = indexes[indexes_last - distance_back - 1] + 1;
        }
      }
    }
    else {++indexes[indexes_last]}
  }
  return result;
}

function products(sets) {
  var result = [];
  var len = sets.length;
  var product;
  for (var i = 0; i < len; ++i) {
    product = 1;
    inner_len = sets[i].length;
    for (var j = 0; j < inner_len; ++j) {
      product *= sets[i][j];
    }
    result.push(product);
  }
  return result;
}

console.log(products(combos([1, 3, 5, 7, 9, 11], 3)));
0 голосов
/ 13 ноября 2018

https://github.com/dankogai/js-combinatorics

Нашел эту библиотеку. Проверено, чтобы работать. Ниже из документа библиотеки:

var Combinatorics = require('js-combinatorics');
var cmb = Combinatorics.combination(['a','b','c','d'], 2);
while(a = cmb.next()) console.log(a);
//  ["a", "b"]
//  ["a", "c"]
//  ["a", "d"]
//  ["b", "c"]
//  ["b", "d"]
//  ["c", "d"]
0 голосов
/ 02 февраля 2016
var create3Combi = function(array) {
    var result = [];
    array.map(function(item1, index1) {
        array.map(function(item2, index2) {
            for (var i = index2 + 1; i < array.length; i++) {
                var item3 = array[i];
                if (item1 === item2 || item1 === item3 || item2 === item3 || index2 < index1) {
                    continue;
                }
                result.push([item1, item2, item3]);
            }
        });
    });

    return result;
};

var multiplyCombi = function(array) {
    var multiply = function(a, b){
        return a * b;
    };

    var result = array.map(function(item, index) {
        return item.reduce(multiply);
    });

    return result;
}

var numArr = [1, 3, 5, 9];

//  create unique 3 number combination
var combi = create3Combi(numArr); //[[1,3,5],[1,3,9],[1,5,9],[3,5,9]]

// multiply every combination
var multiplyResult = multiplyCombi(combi); //[15,27,45,135];
0 голосов
/ 16 марта 2013

Используя узел, вы можете сделать это довольно легко, используя библиотеку. Сначала установите bit-twiddle, используя npm:

npm install bit-twiddle

Тогда вы можете использовать его в своем коде так:

//Assume n is the size of the set and k is the size of the combination
var nextCombination = require("bit-twiddle").nextCombination
for(var x=(1<<(k+1))-1; x<1<<n; x=nextCombination(x)) {
   console.log(x.toString(2))
}

Переменная x является бит-вектором, где бит i устанавливается, если в комбинации содержится i-й элемент.

0 голосов
/ 31 октября 2010

Рекурсивная функция для этого, когда вам нужно выбрать k чисел среди n чисел. Не проверял. Найдите, есть ли ошибка, и исправьте ее: -)

var result = [];

foo(arr, 0, 1, k, n); // initial call

function foo(arr, s, mul, k, n) {
    if (k == 1) {
        result.push(mul);
        return;
    }

    var i;
    for (i=s; i<=n-k; i++) {
        foo(arr, i+1, mul*arr[i], k-1, n-i-1);
    }
}

Это рекурсивная функция.

  1. Первый параметр - массив arr.

  2. Второй параметр - целое число s. Каждый вызов вычисляет значения для части массива, начиная с индекса s. Рекурсивно я увеличиваю s, и поэтому массив для каждого вызова рекурсивно становится меньше.

  3. Третий параметр - это значение, которое вычисляется рекурсивно и передается в рекурсивном вызове. Когда k становится 1, оно добавляется в массив результатов.

  4. k в размере желаемой комбинации. Он рекурсивно уменьшается и, когда становится равным 1, вывод добавляется в массив результатов.

  5. n - размер массива arr. На самом деле n = arr.length

...