Question

使用scala学习函数式编程。碰到了这个练习。

写一个递归函数来得到第n个Fibonacci数（http://mng.bz/C29s）。前两个Fibonacci数是0和1.第n个数总是前两个的总和 - 序列从0,1,1,2,3,5开始。你的定义应该使用本地尾递归函数。

def fib(n: Int): Int

我的答案计算n + 1个值并返回第n个。任何人都可以向我展示一个更好的实现，其中不计算额外的第n + 1值吗？

object patterns {

    def fib(n : Int): Int = {
        @annotation.tailrec
        def go(n: Int, prev2: Int, prev: Int): Int =
            if(n<=0) prev2 
            else go(n-1, prev, prev2+prev)
        go(n, 0, 1)
      }
}

如果有人感兴趣，这是来自Chiusano和Bjarnason的scala函数式编程。练习2.1 期待回复。

Answer 1

我想这个：

def fib2(n: Int): Int = {
  if (n < 1) 0
  else if (n < 2) 1
  else {
    @annotation.tailrec
    def go(i: Int, prev2: Int, prev: Int): Int =
      if (i == n) prev
      else go(i + 1, prev, prev2 + prev)
    go(2, 1, 1)
  }
}

或使用ByName参数：

def fib3(n: Int): Int = {
  @annotation.tailrec
  def go(n: Int, prev2: Int, prev: => Int): Int =
    //                            ^ ByName
    if (n <= 0) prev2
    else {
      val p = prev
      go(n - 1, p, prev2 + p)
    }
  go(n, 0, 1)
}

Answer 2

我喜欢使用stream来实现它。因为代码更短，更容易理解。

      def fibo():Stream[Int] = {
        def addRec(o1:Int,o2:Int):Stream[Int] = {
          o1 #:: addRec(o2,o1 + o2)
        }
        addRec(1,1)
      }

      println(fibo().take(100).toList)

让nth调用fibo.drop(n-1)(0)

很简单

函数式编程：scala中的递归循环输出fibonacci序列

2 个答案: