优化Ruby中的嵌套循环

Question

在Ruby中，我有三个嵌套循环：

array.each do |a|
  array.each do |b|
    array.each do |c|
      puts a * b * c
    end
  end
end

如果嵌套循环的数量可以增加到5-10个或更多的迭代，我如何优化这段代码？

示例：

array.each do |a|
  array.each do |b|
    array.each do |c|
       array.each do |d|
         array.each do |e|
           array.each do |f|
             puts a * b * c * d * e * f
           end
         end
      end
    end
  end
end

Answer 1

你可以这样做：

array.repeated_combination(array.size).each do |combination| 
  puts combination.reduce(:*)
end

Array#repeated_combination返回一个枚举数，该枚举数生成所有可能的组合。

因为此方法在打印任何输出之前生成所有组合可能需要一段时间，这取决于数组的大小。请记住，可能的组合数量增长得相当快：O(nⁿ)和n是数组中的元素数。

Answer 2

以下是另外两种方法(尽管我更喜欢@spickermann的答案)

#1

array = [1,2,3]
n = 4
arr = array.product(*[array]*(n-1)).map { |arr| arr.reduce(:*) }
arr.size #=> 81 = arr.size**n
arr.each { |e| puts e }
1
2
3
2
4
6
3
6
9
...
54
27
54
81

如果您只想打印产品，请将map替换为each，arr.reduce(:*)替换为puts arr.reduce(:*)。

#2

sz = array.size
(0...sz**n).map { |i| i.to_s(sz)
                       .rjust(n,'_')
                       .chars
                       .reduce(1) { |t,e| t * (e=='_' ? 1 : array[e.to_i]) }
                }