编辑代码

object Main{
    def isPrime(n: Int): Boolean = {
  if (n <= 1) {
    false
  } else if (n == 2) {
    true
  } else {
    val limit = math.sqrt(n).toInt
    (2 to limit).forall(i => n % i != 0)
  }
}
    def main(args:Array[String]):Unit={
        println("please enter an interge:")
        val x = scala.io.StdIn.readInt()
        if(isPrime(x)){
            println("prime")
        }else{
            println("not a prime")
        }
    }
}