如何开发一个 Ruby 数组

Question

我有一个 n 维的因式分解数组，我想开发它。

这是一个例子：

develop([:a, :aa]) #=> [[:a, :aa]]

...等同于：[:a].product([:aa])。

或者，更复杂：

develop([:a, [:aa, :bb]]) #=> [[:a, :aa],
                               [:a, :bb]]

我正在使用 Ruby 1.9。谢谢你的任何想法。

编辑：

另一个例子，有 3 层嵌入数组：

develop([:a, [[:b, [:ba, :bb]],
              [:c, [:ca, :cb]],
              [:d, [:da, :db]]]]) #=> [[:a, :b, :ba],
                                       [:a, :b, :bb],
                                       [:a, :c, :ca],
                                       [:a, :c, :cb],
                                       [:a, :d, :da],
                                       [:a, :d, :db]]

我想知道我们是否可以使用 Array 的产品方法 (http://ruby-doc.org/core-1.9.3/Array.html#method-i-product)，即使我们有一些嵌入式数组。

Answer 1

我不确定我是否完全理解您要对这些糟糕的数组执行什么操作，但我设法创建了一个函数，可以为您指定的两种情况提供正确的输出。完整代码如下：

def develop(x)
  return x unless x.is_a? Array
  y = []
  x[1].each do |s|
    d = develop(s)
    d = [d] unless d.is_a? Array
    d.each do |t|
      t = [t] unless t.is_a? Array
      y << [x.first] + t
    end
  end
  return y
end

x = [:a,
        [
            [:b, [:ba, :bb]],
            [:c, [:ca, :cb]],
            [:d, [:da, :db]]
        ]
    ]

p develop(x)

p develop [:a, [:aa, :bb]]

输出是：

C:\Users\David\Documents\scraps\test_ruby>ruby test.rb
[[:a, :b, :ba], [:a, :b, :bb], [:a, :c, :ca], [:a, :c, :cb], [:a, :d, :da], [:a, :d, :db]]
[[:a, :aa], [:a, :bb]]

编辑 1：这是一个较短的版本，也提供了正确的输出：

def develop(x)
  return [x] unless x.is_a? Array
  Array(x.last).collect do |s|
    develop(s).collect do |t|
      [x.first] + Array(t)
    end
  end.flatten 1
end