Scala流以查找特定间隔内的素数

Question

问题：

我需要创建一个 Scala 程序，它使用 Stream 类并从区间 [i, j] 中找到第 n 个素数（而 1

更多信息：

我是 Scala 的新手，但我已经寻找了有关如何使用 Stream 在 Scala 中查找素数的各种示例。他们都没有帮助我实现我的目标。

我似乎无法理解如何使流成为区间 [i, j] 中的有限列表以及如何从该区间中获取第 n 个素数。

到目前为止我的代码：

def primeStream(args: Array[String], s: Stream[Int]): Stream[Int] =
        Stream.cons(s.head, primeStream(args,s.tail filter {_ % s.head != 0 }))

if (args(0).toInt < 1) {
        println("Entered parameter i does not meet requirements 1<=i<j (1<=" + args(0) + "<" + args(1) + ")")
        sys.exit(1)
} else if (args(1).toInt < args(0).toInt) {
        println("Entered parameter j does not meet requirements 1<=i<j (1<=" + args(0) + "<" + args(1) + ")")
        sys.exit(1)
}

val primes = primeStream(args,Stream.from(args(0).toInt)) // start from i element

primes take args(1).toInt foreach println //Take j elements

任何帮助将不胜感激！

解决方案：

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

if (args(0).toInt < 1) {
        println("Entered parameter i does not meet requirements 1<=i<j (1<=" + args(0) + "<" + args(1) + ")")
        sys.exit(1)
} else if (args(1).toInt < args(0).toInt) {
        println("Entered parameter j does not meet requirements 1<=i<j (1<=" + args(0) + "<" + args(1) + ")")
        sys.exit(1)
} else if (args(0).toInt == 1) {
        println("1 is not a prime by definition!") 
        sys.exit(1)  // if args(0) is 1 then function hangs up - didn't come up with a better solution for this
}

val primes = primeStream(Stream.from(args(0).toInt)) // get primes starting from given parameter

println(primes.takeWhile( _ < args(1).toInt).take(args(2).toInt).last) // get n-th prime and print it out

Answer 1

您只需要在满足特定条件时让您的流生成值：

primes takeWhile(_ < j) take n foreach println

当然，您需要正确使用primeStream 函数。

算法部分，你最好在stackoverflow上搜索：

• Calculating prime numbers in Scala: how does this code work?

Answer 2

这是一个学习练习，还是您在生产中需要它？对于后者，我建议使用来自spire 库的 spire.math.prime.stream。它使用 Eratosthenes 的分段流 实现，这可能比您在短时间内自己想出的要好。它还使用任意精度整数，因此适用于大于 2^64 的数字。

scala> import spire.math._
import spire.math._

scala> prime.stream.drop(10).take(10).toArray
res16: Array[spire.math.SafeLong] = Array(31, 37, 41, 43, 47, 53, 59, 61, 67, 71)