我有以下递归函数
trait SequenceGenerator[T] {
def program(l: List[T])(implicit rule: ProductionRule[T]): List[T] = {
l.flatMap(rule.generate)
}
def sequenceNumber(seed: List[T], number: Int)(implicit rule: ProductionRule[T]): List[T] = {
number match {
case 1 => program(seed)
case a => program(sequenceNumber(seed, a - 1))
}
}
}
Run Code Online (Sandbox Code Playgroud)
我想不出一种使sequenceNumber尾递归的方法。
将方法转换为tailrec的想法是,将每个下一个计算步骤放入累加器,然后将该累加器传递回该方法,以便在下一步中,您可以返回累加器,或者根据您的步骤修改后的新累加器再次调用此方法。
因此,您将获得如下内容:
trait SequenceGenerator[T] {
def program(l: List[T])(implicit rule: ProductionRule[T]): List[T] = l.flatMap(rule.generate)
def sequenceNumber(seed: List[T], number: Int)(implicit rule: ProductionRule[T]): List[T] = {
@tailrec
def tailRec(number: Int, acc: List[T]): List[T] = number match {
case 1 => acc
case a => tailRec(a-1,program(acc))
}
tailRec(number,program(seed))
}
}
Run Code Online (Sandbox Code Playgroud)
但是,仅FYI,还有一种用于构建尾递归的更高级的技术称为Trampoling。scala.util.control.TailCalls例如,可以使用它来实现。
我在这项技术上不是很好,但是看起来会像这样:
import TailCalls._
trait SequenceGenerator[T] {
def program(l: TailRec[List[T]])(implicit rule: ProductionRule[T]):TailRec[List[T]] = {
l.map(_.flatMap(rule.generate))
}
def sequenceNumber(seed: TailRec[List[T]], number: TailRec[Int])(implicit rule: ProductionRule[T]): TailRec[List[T]] = {
number flatMap {
case 1 => tailcall(program(seed))
case a => tailcall(program(sequenceNumber(seed, done(a - 1))))
}
}
}
Run Code Online (Sandbox Code Playgroud)
sequenceNumber此处不能标记为tailrec,但实际上是“背景”中的tailrec。要获得真实的计算结果,您需要致电sequenceNumber(done(...),done(...)).result