Scalaでエラトステネスのふるい

参考：wikipedia:エラトステネスの篩

import scala.collection.mutable

object Main{
  def main(args: Array[String]): Unit = {
    val N = 120 // N以下の素数
    val primes = mutable.BitSet()

    primes(2) = true

    // すべての数を素数として初期化。2以外の偶数は素数にはなりえないので除外。
    for (i <- 3 to N by 2) {
      primes(i) = true
    }

    // ふるいにかける。
    for (i <- 3 to Math.sqrt(N).toInt by 2 if primes(i); j <- i * 2 to N by i) {
      primes(j) = false
    }

    println(primes.mkString(","))
    println(primes.drop(9).head) // 10番目の素数
  }
}