[Swift Algorithm] 소수 만들기 Programmers(Lv.1)

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

 

제한 조건

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

 

입출력 예

nums	result
[1,2,3,4]	1
[1,2,7,6,4]	4

 

내 문제 풀이

import Foundation

func solution(_ nums:[Int]) -> Int {
    var sum = [Int]()
    var count = 0
    
    for i in 0..<nums.count-2 {
        for j in i+1..<nums.count {
            for k in j+1..<nums.count {
                sum.append(nums[i] + nums[j] + nums[k])
            }
        }
    }
    
    for i in 0..<sum.count {
        for j in 2..<sum[i] {
            if sum[i] % j == 0 {
                break
            }
            else if j == sum[i]-1 && sum[i] % j != 0 {
                count += 1
            }
        }
    }

    return count
}

• 3중 for문을 사용하여 모든 3개 수의 합을 구했고, sum 배열에 append 해주었다.
• 2중 for문 이용하여 소수인 수의 개수를 구해주었다.
• 2부터 해당 수의 전까지 나눠보며 끝까지 나눠지지 않은 수는 소수로 판별하여 count에 개수를 더해주었다.

 

💡 피드백

• 코드가 길어지니 소수 구하는 부분은 따로 함수로 만들어줬어도 좋을 것 같다.
• 이전에 소수 구하는 문제를 풀었던 기억이 있어 큰 고민 없이 구현할 수 있었다.

 

---

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12977