用Scala语言判断输入的整数是否为素数代码

下面是用Scala语言判断输入的整数是否为素数的代码：

import scala.math._

def isPrime(num: Int): Boolean = {
  if(num <= 1) {
    false
  } else if(num == 2) {
    true
  } else {
    val limit = sqrt(num).toInt
    for(i <- 2 to limit) {
      if(num % i == 0) {
        return false
      }
    }
    true
  }
}

println(isPrime(7)) // true
println(isPrime(21)) // false

这段代码定义了一个isPrime函数，参数为一个整数num，返回值为Boolean类型。在函数内部，先判断num是否小于等于1，若是，则返回false。再判断num是否等于2，若是，则返回true。然后，计算num的平方根，取整数，作为循环上限limit。接着，从2循环到limit，查看是否存在num的约数，若存在，则返回false。最后，如果没有找到约数，则返回true。