ruby - 如何在 Ruby 中编写递归阶乘函数？

Question

我只是想得到一些关于如何在 Ruby 中编写递归阶乘函数的帮助。我有以下 lisp 代码，但我想在 Ruby 中做同样的事情。

(defun factorial (N)
    (if (= N 1) 1
        (* N (factorial (- N 1)))))

Answer 1

以下是用 ruby​​ 编写代码的方法:

def factorial(n)
  return 1 if n == 1
  n * factorial(n - 1)
end

factorial(5)
#=> 120
factorial(7)
#=> 5040

编辑 Stefan 的评论:

要避免 SystemStackError 错误与较大的 n 值，请使用 tail-recursive 方法。还必须启用 Ruby 的 tailcall 优化。

# before edit
factorial(100_000).to_s.size
#=> stack level too deep (SystemStackError)

避免SystemStackError

RubyVM::InstructionSequence.compile_option = {
  tailcall_optimization: true,
  trace_instruction: false
}

RubyVM::InstructionSequence.new(<<-CODE).eval
  def factorial(n, acc = 1)
    return acc if n == 1
    factorial(n - 1, n * acc)
  end
CODE

puts factorial(100_000).to_s.size
#=> 456574

Resource 1 Resource 2