Question

我试图记住我对Pascal三角形生成器的实现，作为一个Ruby学习实验。我有以下工作代码：

module PascalMemo
  @cache = {}
  def PascalMemo::get(r,c)
    if @cache[[r,c]].nil? then 
      if c == 0 || c == r then 
        @cache[[r,c]] = 1
      else
        @cache[[r,c]] = PascalMemo::get(r - 1, c) + PascalMemo::get(r - 1, c - 1)
      end
    end
    @cache[[r,c]]
  end
end

def pascal_memo (r,c)
  PascalMemo::get(r,c)
end

这可以更简洁吗？具体来说，我可以创建一个全局范围的函数，其局部闭包比这更简单吗？

Answer 1

def pascal_memo
  cache = [[1]]
  get = lambda { |r, c|
    ( cache[r] or cache[r] = [1] + [nil] * (r - 1) + [1] )[c] or
      cache[r][c] = get.(r - 1, c) + get.(r - 1, c - 1)
  }
end

p = pascal_memo
p.( 10, 7 ) #=> 120

请注意，上面的构造确实实现了memoization，它不仅仅是一个简单的递归方法。

Answer 2

这可以更简洁吗？

看起来很清楚，IMO和module通常是一种很好的直觉。

我可以创建一个全局范围的函数，其局部闭包比这更简单吗？

另一种选择是递归lambda：

memo = {}

pascal_memo = lambda do |r, c|
  if memo[[r,c]].nil?
    if c == 0 || c == r
      memo[[r,c]] = 1
    else
      memo[[r,c]] = pascal_memo[r - 1, c] + pascal_memo[r - 1, c - 1]
    end
  end
  memo[[r,c]]
end

pascal_memo[10, 2]
# => 45

Answer 3

我找到了一种方法来实现我想要的更令人满意的方法，因为它产生的是一个函数而不是一个lambda：

class << self
  cache = {}

  define_method :pascal_memo do |r,c|
    cache[[r,c]] or 
    (if c == 0 or c == r then cache[[r,c]] = 1 else nil end) or 
    cache[[r,c]] = pascal_memo(r-1,c) + pascal_memo(r-1,c-1)
  end
end

这将打开主对象的元类/单例类，然后使用define_method添加一个关闭缓存变量的新方法，该方法超出了除pascal_memo方法之外的所有内容。

带有memoization的Ruby Pascal三角形生成器

3 个答案: