Scala将递归函数与内部映射转换为尾递归

Question

我有一个对象，我们称它为 RT，它有两个字段：一个字符串和一个自身的 List。

def case class RT[String](topic: String, children: List[RT[String]])

现在我有了以下递归函数：

def func(tree: RT[String]): List[String] = {
  tree.children match  {
    case Nil=> List(tree.topic)
    case child =>
      val subTrees = child.map(rt => func(rt))
      subTrees.foldLeft(List(tree.topic))((acc,list) => acc ::: list ::: List(tree.topic))
  }
}

这让我得到了我想要的。但是，在非常嵌套的情况下，我得到了一个 StackOverflow，所以我希望将前面的函数转换成它的尾递归形式。但是，我看不出如何。有可能吗？

Answer 1

首先，让我简化你的原始代码，很多：

case class RT(topic: String, children: List[RT])

def func(tree: RT): List[String] =
  tree.topic :: tree.children.flatMap(rt => func(rt))

(借用 Inigo Montoya，我不认为 case class RT[String] 意味着你认为的意思。它不是以某种方式构建在 Scala 字符串上的参数化类。它是构建在任意类型上的参数化类您正在调用“String”，尽管它不一定具有任何类似字符串的特性。稍后，您将使用真正的字符串实例化类型类。）

它仍然不是尾递归的，所以让我们玩通常的把戏：

def func(tree: RT): List[String] = {
  @tailrec
  def funcAcc(trees: List[RT], acc: List[String]): List[String] = {
    if (trees.isEmpty)
      acc
    else
      funcAcc(trees.tail ::: trees.head.children, trees.head.topic :: acc)
  }
  funcAcc(List(tree), Nil).reverse
}

那里，行得通。

（我们本可以避免使用reverse，但这需要追加到不断增长的列表中，并且一大堆追加通常比相同数量的前置要慢得多，然后反过来。）