在Scala中计算素数流的代码

object PrimeNumbers {
  def isPrime(n: Int): Boolean = {
    if (n <= 1) false
    else if (n == 2) true
    else !(2 to (Math.sqrt(n).toInt + 1)).exists(x => n % x == 0)
  }

  def primes: Stream[Int] = {
    def generatePrimes(n: Int): Stream[Int] = {
      if (isPrime(n)) n #:: generatePrimes(n + 1)
      else generatePrimes(n + 1)
    }

    generatePrimes(2)
  }

  def main(args: Array[String]): Unit = {
    val primeStream = primes.take(10) // 取前10个素数
    primeStream.foreach(println)
  }
}