16 lines
470 B
Scala
16 lines
470 B
Scala
def isKPrime(n: Int, k: Int, d: Int = 2): Boolean = (n, k, d) match {
|
|
case (n, k, _) if n == 1 => k == 0
|
|
case (n, _, d) if n % d == 0 => isKPrime(n / d, k - 1, d)
|
|
case (_, _, _) => isKPrime(n, k, d + 1)
|
|
}
|
|
|
|
def kPrimeStream(k: Int): Stream[Int] = {
|
|
def loop(n: Int): Stream[Int] =
|
|
if (isKPrime(n, k)) n #:: loop(n+ 1)
|
|
else loop(n + 1)
|
|
loop(2)
|
|
}
|
|
|
|
for (k <- 1 to 5) {
|
|
println( s"$k: [${ kPrimeStream(k).take(10) mkString " " }]" )
|
|
}
|