在紅寶石

Question

創建從一個多維陣列排列

我有紅寶石以下多維數組：

[[1,2], [3], [4,5,6]] 

我需要有下面的輸出：

[[1,3,4], [1,3,5], [1,3,6], [2,3,4], [2,3,5], [2,3,6]] 

我試圖創建遞歸函數，但我沒有太多的運氣。

是否有任何Ruby功能可以幫助解決這個問題？或者是遞歸執行它的唯一選擇？

感謝

Answer 1

是啊，Array#product做到了這一點（Cartesian product）：

a = [[1,2], [3], [4,5,6]] 
head, *rest = a # head = [1,2], rest = [[3], [4,5,6]] 
head.product(*rest) 
#=> [[1, 3, 4], [1, 3, 5], [1, 3, 6], [2, 3, 4], [2, 3, 5], [2, 3, 6]] 

另一個變化：

a.inject(&:product).map(&:flatten) 
#=> [[1, 3, 4], [1, 3, 5], [1, 3, 6], [2, 3, 4], [2, 3, 5], [2, 3, 6]]