[Swift] 프로그래머스 Lv.1 : 소수 만들기

▲문제 link🔗

 

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때,

nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

---

제한사항

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

---

입출력 예

 

입출력 예 설명

입출력 예 #1

• [1,2,4]를 이용해서 7을 만들 수 있습니다.

입출력 예 #2

• [1,2,4]를 이용해서 7을 만들 수 있습니다.
• [1,4,6]을 이용해서 11을 만들 수 있습니다.
• [2,4,7]을 이용해서 13을 만들 수 있습니다.
• [4,6,7]을 이용해서 17을 만들 수 있습니다.

 

---

 

✏️ 풀이

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

 

소수인지 판별하는 함수
👉 2부터 해당 숫자보다 작은 수까지 반복
👉 하나라도 나누어떨어지면 소수가 아닌 것으로 판별

 

func solution(_ nums:[Int]) -> Int {
    var answer = 0
    
    for i in 0...nums.count - 3 {
        let first = nums[i]
        for j in i+1...nums.count - 2 {
            let second = nums[j]
            for k in j+1...nums.count - 1 {
                let third = nums[k]
                if isPrime(first + second + third) {
                    answer += 1
                }
            }
        }
    }
    return answer
}

우선 nums에 중복된 숫자가 없다는 제한 사항이 있기 때문에 중복 조합 확인 코드는 없다.

세 개의 중첩 반복문을 사용하여
1. 배열의 첫번째 요소 ~ 뒤에서 세번째 요소
2. 선택된 요소의 다음 요소 ~ 뒤에서 두번째 요소
3. 2번의 선택된 요소의 다음 요소 ~ 마지막 요소
이렇게 반복하여 각각의 요소를 하나씩 선택 후 각 반복문에서 선택된 숫자의 합을 확인했다.

중복된 조합을 생성하지 않는 이유는 첫번째 숫자는 배열의 마지막 두 숫자 중 하나가 될 수 없다.
두 번째 숫자는 첫 번째 선택된 숫자 이후의 요소 중 하나이고 마지막 숫자는 선택될 수 없다.
세 번째 숫자는  앞의 두 숫자 이후부터 마지막 요소 중 하나이다.
앞서 말했듯이 제한사항에 배열 안에 중복된 숫자가 없고, 세 숫자가 모두 다르게 선택되기 때문에 중복조합이 발생하지 않는다!