獲取多維數組的N個元素的所有組合

Question

我正在嘗試編寫一種算法，以獲取M元素的多維數組中N元素的所有可能組合。

就像是：

function getCombinations(arr, n){
  ...
}

var arr = [ ["A"],["B","C"],["D","E"]];
var n = 2;

getCombinations(arr,n);

這應該產生：

[
["A","B"],["A","C"],["A","D"],["A","E"],
["B","D"],["B","E"],
["C","D"],["C","E"]
]

數組中元素的數量可以變化，唯一要設置的是組合中元素的數量。

順序無關緊要，但是您不能重復，我的意思是["A","B"] == ["B","A"] ，因此第二個沒有考慮。

有什么幫助嗎？

Answer 1

ChrisB解決方案有一個錯誤，他沒有在arr.shift之前緩存循環的長度，並且沒有返回最后一個組合，我認為這可以完成工作：

function getCombinations (arr, n) {
    var i, j, k, elem, l = arr.length, childperm, ret = [];
    if (n == 1){
        for (i = 0; i < arr.length; i++) {
            for (j = 0; j < arr[i].length; j++) {
                ret.push([arr[i][j]]);
            }
        }
        return ret;
    }
    else {
        for (i = 0; i < l; i++) {
            elem = arr.shift();
            for (j = 0; j < elem.length; j++) {
                childperm = getCombinations(arr.slice(), n-1);
                for (k = 0; k < childperm.length; k++) {
                    ret.push([elem[j]].concat(childperm[k]));
                }
            }
        }
        return ret;
    }
}


getCombinations([["A"],["B"],["C","D"]], 2);

// [["A", "B"], ["A", "C"], ["A", "D"], ["B", "C"], ["B", "D"]]

getCombinations([["A"],["B"],["C"],["D"]], 2);

// [["A", "B"], ["A", "C"], ["A", "D"], ["B", "C"], ["B", "D"], ["C", "D"]]

Answer 2

更新

根據您限制不能合並包含在開頭的同一數組中的元素，我將算法修改為以下內容：

這是更新的jsfiddle，現在甚至可以以正確的格式輸出數據：） http://jsfiddle.net/QKg2H/7/

function getCombinations(arr, n)
{
    if(n == 1)
    {
        var ret = [];
        for(var i = 0; i < arr.length; i++)
        {
            for(var j = 0; j < arr[i].length; j++)
            {
                ret.push([arr[i][j]]);
            }
        }
        return ret;
    }
    else
    {
        var ret = [];
        for(var i = 0; i < arr.length; i++)
        {
            var elem = arr.shift();
            for(var j = 0; j < elem.length; j++)
            {
                var childperm = getCombinations(arr.slice(), n-1);
                for(var k = 0; k < childperm.length; k++)
                {
                    ret.push([elem[j]].concat(childperm[k]));
                }
            }
        }
        return ret;
    }
}

該算法仍然是遞歸的，但是現在它將依次考慮每個第二度元素，但不會相互考慮。 除此之外，它仍然彈出一個元素，然后附加所有剩余元素的排列。 我希望這很簡單。