递归到scala中的Tail-recursive

时间:2018-05-21 01:59:02

标签: scala recursion tail-recursion

我有以下代码

object TailRec
{
    def func1(n:Int) : Int =
    {
        if (n < 10)
            func2(n)
        else
            func1(n/10) + func2(n % 10)
    }

    def func2(n: Int) : Int =
    {
        @tailrec def _func2(n: Int, result: Int): Int =
        {
            if (n <= 1)
                result
            else
                _func2(n-1,n*result)
        }
        _func2(n,1)
    }

    def test(n: Int) : Boolean =
    {
        if (n > 2)
            n == func1(n)
        else
            false
    }

}

我设法重写func2,但我不太确定如何转换bool函数。我考虑过匹配和大小写,但我仍然需要调用func1来获取结果进行比较。另一个问题是如何在func1本身中分解双重函数调用。任何提示?

1 个答案:

答案 0 :(得分:3)

您已将func2转换为尾递归形式。 类似地,使用额外的累加器,您可以将func1转换为尾递归形式(我修改了func1的接口,但您也可以创建另一个内部辅助函数)。由于test不是递归的,因此这里不需要做任何事情。

import scala.annotation.tailrec

object TailRec {
  @tailrec def func1(n:Int, acc: Int): Int = {
    if (n < 10) func2(n)
    else func1(n / 10, acc + func2(n % 10))
  }

  def func2(n: Int): Int = {
    @tailrec def _func2(n: Int, result: Int): Int = {
      if (n <= 1) result
      else _func2(n-1,n*result)
    }
    _func2(n,1)
  }

  def test(n: Int): Boolean = (n > 2) && (n == func1(n, 0))
}