scala - 如何在 Scala 中使用滑动流获取斐波那契数列？

Question

在 Stream 文档中有一个很好的例子可以获取斐波那契数。

val fibs:Stream[Int] = 0 #:: 1 #:: fibs.zip(fibs.tail).map { n => n._1 + n._2 }

我想通过使用滑动来实现它，所以我尝试了以下方法。

val test = 0 #:: 1 #:: Stream.empty
test.sliding(2).map(_.sum).toStream

最后一行正确获取 Stream(1, ?) 但是当我按如下方式将其连接到上面时，当我尝试获取第三个成员时出现错误(可能是堆栈溢出，我看不到确切的错误消息，因为它太长了) .

val fibs2:Stream[Int] = 0 #:: 1 #:: fibs2.sliding(2).map(_.sum).toStream

如果我给出如下 3 个数字，它会计算前两个数字的总和。但这不是斐波那契数。

val fibs3:Stream[Int] = 0 #:: 0 #:: 1 #:: fibs3.sliding(2).map(_.sum).toStream

任何想法或帮助将不胜感激。

更新

我怀疑错误的原因是滑动方法返回Iterator，需要使用hasNext方法知道下一个值是否可用

如果给定第一个种子，滑动方法应该计算前 n 个数字的任何总和，称为 tribonacci (n=3)、tetranacci (n=4) 等。

Answer 1

问题似乎是GroupedIterator (由 sliding 返回)过于急切。创建每个滑动窗口时，它会强制当前窗口之后的下一个元素。

这是一个简单的例子:

import scala.util.Try

def bad[T]: Stream[T] = throw new RuntimeException("Don't peek!")

// Should be able to view group of first 2 elements without error,
// but sliding and grouped both read the 3rd element
def testA: Stream[Int] = 1 #:: 2 #:: bad

Try { testA.sliding(2).next }
// res0: scala.util.Try[scala.collection.immutable.Stream[Int]] = Failure(java.lang.RuntimeException: Don't peek!)

Try { testA.grouped(2).next }
// res1: scala.util.Try[scala.collection.immutable.Stream[Int]] = Failure(java.lang.RuntimeException: Don't peek!)

// Adding an extra element before the bad entry gives
// sufficient padding for a sliding window of 2
def testB: Stream[Int] = 1 #:: 2 #:: 3 #:: bad

Try { testB.sliding(2).next }
// res2: scala.util.Try[scala.collection.immutable.Stream[Int]] = Success(Stream(1, ?))

Try { testB.grouped(2).next }
// res3: scala.util.Try[scala.collection.immutable.Stream[Int]] = Success(Stream(1, ?))

而不是 sliding ，您可以使用 scanLeft :

val fibs: Stream[Int] = 0 #:: fibs.scanLeft(1)(_+_)

scan功能有点像 fold ，但会产生所有中间结果。所以你得到的是这样的:

0

1 = 1

0 + 1 = 1

1 + 1 = 2

1 + 2 = 3

2 + 3 = 5

...