JavaScript数组 - 查找2个元素的组合数

时间:2015-03-20 14:33:24

标签: javascript arrays

我来自Ruby背景,其中包含enumerable类。在Ruby中,我可以很容易地找到数组元素的组合。

array.combination(2).count

我知道JavaScript没有这样的内置函数,所以我想知道如何在JS中实现它。我在想像

我有一个数组如下

var numbers = [9,7,12]
var combos = []
for (var i = 0; i < numbers.length; i++)  {
  combos.push([numbers[i], numbers[i+1])
}

顺便说一下,可能的组合是

[9,7], [9,12] and [7,12]

因此,通过调用此数组上的length函数,将返回3。

有什么想法吗?

3 个答案:

答案 0 :(得分:3)

怎么样:

for (var i = 0; i < numbers.length; i++)
    for (var j = i + 1; j < numbers.length; j++)
        combos.push([numbers[i], numbers[j]]);

答案 1 :(得分:1)

您是否严格谈论阵列的2种组合,或者您对k组合解决方案感兴趣?

在此gist

中找到此信息
function k_combinations(set, k) {
var i, j, combs, head, tailcombs;

if (k > set.length || k <= 0) {
    return [];
}

if (k == set.length) {
    return [set];
}

if (k == 1) {
    combs = [];
    for (i = 0; i < set.length; i++) {
        combs.push([set[i]]);
    }
    return combs;
}

// Assert {1 < k < set.length}

combs = [];
for (i = 0; i < set.length - k + 1; i++) {
    head = set.slice(i, i+1);
    tailcombs = k_combinations(set.slice(i + 1), k - 1);
    for (j = 0; j < tailcombs.length; j++) {
        combs.push(head.concat(tailcombs[j]));
    }
}
return combs;
}

答案 2 :(得分:0)

这是一个递归函数,它适用于任何数字:

function combination(arr, num) {
  var r= [];

  for(var i = 0 ; i < arr.length ; i++) {
    if(num===1) r.push([arr[i]]);
    else {
      combination(arr.slice(i+1), num-1).forEach(function(val) {
        r.push([].concat(arr[i], val));
      });
    }
  }
  return r;
} //combination

Working Fiddle