遍历 Stream答案

【问题标题】：Iterating through Stream遍历 Stream
【发布时间】：2017-12-27 12:47:22
【问题描述】：

我对 Scala 非常陌生，我必须找到 [i;j] 范围内数字的质数。这就是我填充流的方式：

val stream = (i to j).toStream

然后我调用这个应该找到素数的函数：

  def findPrimes(s: Stream[Int]): Stream[Int] = {
    println("HEAD: " + s.head)
    return findPrimes(s.tail)
//    s.head #:: findPrimes(s.tail.filter( _ % s.head != 0 ))
  }

将流输出到控制台，我很惊讶为什么会出现异常：

Exception in thread "main" java.util.NoSuchElementException: head of empty stream
    at scala.collection.immutable.Stream$Empty$.head(Stream.scala:1104)
    at scala.collection.immutable.Stream$Empty$.head(Stream.scala:1102)
    at com.example.anna.app.HelloWorld$.findPrimes(HelloWorld.scala:43)
STREAM SIZE IS 100
    at com.example.anna.app.HelloWorld$.prime(HelloWorld.scala:32)
HEAD: 1
    at com.example.anna.app.HelloWorld$.delayedEndpoint$com$example$anna$app$HelloWorld$1(HelloWorld.scala:11)
HEAD: 2
HEAD: 3
    at com.example.anna.app.HelloWorld$delayedInit$body.apply(HelloWorld.scala:3)
HEAD: 4
    at scala.Function0.apply$mcV$sp(Function0.scala:34)
HEAD: 5
HEAD: 6
    at scala.Function0.apply$mcV$sp$(Function0.scala:34)
HEAD: 7
    at scala.runtime.AbstractFunction0.apply$mcV$sp(AbstractFunction0.scala:12)
HEAD: 8
HEAD: 9
    at scala.App.$anonfun$main$1$adapted(App.scala:76)
HEAD: 10
HEAD: 11
    at scala.collection.immutable.List.foreach(List.scala:389)
HEAD: 12
    at scala.App.main(App.scala:76)
HEAD: 13
    at scala.App.main$(App.scala:74)
HEAD: 14
HEAD: 15
    at com.example.anna.app.HelloWorld$.main(HelloWorld.scala:3)
HEAD: 16
    at com.example.anna.app.HelloWorld.main(HelloWorld.scala)
HEAD: 17

从 17 开始到结束，没有一个例外。而且这条注释行也不起作用 - 它会引发相同的异常。

任何帮助和建议将不胜感激。

【问题讨论】：

标签： scala stream

【解决方案1】：

您传递给findPrimes 的流大小有限，并且在您调用findPrimes 时会耗尽。

使用Stream.from 方法创建无限流。

Scala REPL

scala> :paste
// Entering paste mode (ctrl-D to finish)


def primeStream(s: Stream[Int]): Stream[Int] =
    Stream.cons(s.head, primeStream(s.tail filter { _ % s.head != 0 }))
val primes = primeStream(Stream.from(2))


// Exiting paste mode, now interpreting.

primeStream: (s: Stream[Int])Stream[Int]
primes: Stream[Int] = Stream(2, ?)

scala> primes.take(100).toList
res0: List[Int] = List(2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 419, 421, 431, 433, 439, 443, 449, 457, 461, 463, 467, 479, 487, 491, 499, 503, 509, 521, 523, 541)

您的代码有问题。

scala> val primes = primeStream((1 to 100).toStream)
primes: Stream[Int] = Stream(1, ?)

scala> primes.take(100).toList
java.util.NoSuchElementException: head of empty stream
  at scala.collection.immutable.Stream$Empty$.head(Stream.scala:1104)
  at scala.collection.immutable.Stream$Empty$.head(Stream.scala:1102)
  at .primeStream(<console>:12)
  at .$anonfun$primeStream$1(<console>:12)
  at scala.collection.immutable.Stream$Cons.tail(Stream.scala:1169)
  at scala.collection.immutable.Stream$Cons.tail(Stream.scala:1159)
  at scala.collection.immutable.Stream.$anonfun$take$2(Stream.scala:789)
  at scala.collection.immutable.Stream$Cons.tail(Stream.scala:1169)
  at scala.collection.immutable.Stream$Cons.tail(Stream.scala:1159)
  at scala.collection.generic.Growable.loop$1(Growable.scala:54)
  at scala.collection.generic.Growable.$plus$plus$eq(Growable.scala:58)
  at scala.collection.generic.Growable.$plus$plus$eq$(Growable.scala:50)
  at scala.collection.mutable.ListBuffer.$plus$plus$eq(ListBuffer.scala:186)
  at scala.collection.mutable.ListBuffer.$plus$plus$eq(ListBuffer.scala:44)
  at scala.collection.TraversableLike.to(TraversableLike.scala:590)
  at scala.collection.TraversableLike.to$(TraversableLike.scala:587)
  at scala.collection.AbstractTraversable.to(Traversable.scala:104)
  at scala.collection.TraversableOnce.toList(TraversableOnce.scala:294)
  at scala.collection.TraversableOnce.toList$(TraversableOnce.scala:294)
  at scala.collection.AbstractTraversable.toList(Traversable.scala:104)
  ... 28 elided

【讨论】：

但我怎样才能限制 Stream 并将数字仅放在 [i; j] ?万一 - 我不需要 100 个第一个素数，我需要找到范围内的所有素数并返回第 n 个。
如果你放无限流，你可以覆盖所有情况。无限流实际上是有限的。它不会占用任何额外的内存。

【解决方案2】：

为什么这会引发异常？原因是println语句中空流调用的head方法。

所以我们可以对这个流进行模式匹配以在尾部为空时返回空流，如下所示（有更好的方法来遍历流，但我尽量不要对你的代码做太多改动）：

def findPrimes(s: Stream[Int]): Stream[Int] = s match {
         case Stream.Empty =>
            println("END") //consider removing println statements
            Stream.Empty
         case h #:: tl =>
            println(s"HEAD: $h")
            h #:: findPrimes(tl)
     }

【讨论】：