프로그래머스 - 소수 만들기

import Foundation


func check(_ n: Int) -> Bool {
    if n <= 1 { return false }
    if n == 2 { return true }
    if n % 2 == 0 { return false }
    for i in stride(from: 3, through: Int(sqrt(Double(n))), by: 2) {
        if n % i == 0 {
            return false
        }
    }
    return true
}

func solution(_ nums:[Int]) -> Int {
    var answer = 0
    
    for i in 0..<nums.count {
        for j in i+1..<nums.count {
            for k in j+1..<nums.count {
                if i != j && j != k {
                    if check(nums[i] + nums[j] + nums[k]) {
                        answer += 1
                    }
                }
            }
        }
    }
    return answer
}