34일차 TIL

TIL

34일차 TIL

h_luz 2024. 4. 15. 22:04

Swift 알고리즘 및 언어공부

58. 소수 만들기

import Foundation

func solution(_ nums:[Int]) -> Int {

    var answer = 0
    var result = 0

    for a in 0..<nums.count {
        for b in a+1..<nums.count {
            for c in b+1..<nums.count {
                result = nums[a] + nums[b] + nums[c]
                if isPrime(num: result) {
                    answer += 1
                }
            }
        }
    }
    return answer
}

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

소수 구하는 공식이나 함수가 있을까 해서 찾아봤는데, 다들 함수로 만들어서 사용하길래 가져왔다!

소수만 구하면 쉬운 문제였던 것 같음!

우선 1,2,3은 무조건 소수이기 때문에 4보다 작은 수라면 소수.

그리고 num % i 즉, 2 이상인 수로 나눴을 때 0이면 소수가 아니기 때문에 false를 리턴하고, 0이 아닐 경우 소수로 쳐서 true를 리턴.

sqrt는 전에 공부했던 제곱근 함수이다. 즉, 소수를 구하기 위해서는 1을 제외한 √num까지의 수 중에 약수가 있는지 확인하는 것

//다른사람의풀이
import Foundation

func solution(_ nums:[Int]) -> Int {
    var result = [Int]()
    for i in 0..<nums.count {
        for j in i+1..<nums.count {
            for k in j+1..<nums.count {
                result.append(nums[i]+nums[j]+nums[k])
            }
        }
    }
    return result.filter { i in
        for j in 2..<i {
            if i % j == 0 {
                return false
            }
        }
        return true
    }.count
}

이건 다른 사람의 풀이인데, 함수로 구하지 않고,

간단하게 '확인하려는 숫자를 2부터 차례차례 나눠서 나머지가 0이 나오지 않는 숫자' 들을 filter 함수로 골라내서 .count로 개수를 구하는 방식이었다. 조금 더 생각해 보면 이런 식으로도 풀 수 있구나 싶었다.. 난 언제쯤 고차함수를 적절하게 사용할 수 있게 될까.. ㅜ