[Swift] 소수 찾기

import Foundation

func isPrime(_ n: Int) -> Bool {
    if (n < 4) {
        return n == 1 ? false : true
    }
    for i in 2...Int(sqrt(Double(n))) {
        if (n % i == 0) { return false }
    }
    return true
}

func solution(_ n:Int) -> Int {
    return (2...n).filter { isPrime($0) }.count
}