ruby - 用于生成排列的惯用 ruby ？

Question

我想知道这个用于生成排列的函数的惯用版本在 Ruby 中会是什么样子。我知道 [1,2,3].permutation.to_a 会生成相同的结果，但我更感兴趣的是学习 Ruby 以及如何在 Ruby 中处理这样的递归问题。

def permutations(seq)
    if seq.empty? || seq.count == 1
        seq
    else
        seq.map { |x|
            permutations(seq.select { |e| e != x }).map { |p|
                if p.class == Fixnum
                    [x, p]
                else
                    p.unshift(x)
                end
            }
        }.flatten(1)
    end
end

谢谢!

Answer 1

class Array
  def permutations
    return [self] if size < 2
    perm = []
    each { |e| (self - [e]).permutations.each { |p| perm << ([e] + p) } }
    perm
  end
end 

[1, 2, 3].permutations #=> [[1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], [3, 2, 1]] 

来源:http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/32844

编辑:为避免猴子修补，将其放入模块中:

module ArrayExtensions
  def permutations
    #snip
  end
end

Array.send :include, ArrayExtensions