【Swift】素数判定

素数の場合はtrueを返しそれ以外はfalseを返す

func isPrimeNumber(number: Int) -> Bool {

    if number == 1 || number % 2 == 0 {
        return false
    }

    if number == 2 || number == 3 {
        return true
    }

    let sqrtNum = Int(sqrt(Double(number)))

    for i in 2...sqrtNum {
        if number % i == 0 {
            return false
        }
    }
    return true
}

isPrimeNumber(number: 71) //true