문제 설명

주어진 숫자 중 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

입출력 예 설명

입출력 예 #1
[1,2,4]를 이용해서 7을 만들 수 있습니다.

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

내 문제 풀이

• 지난 번 프로그래머스 Level1 두 개 뽑아서 더하기 에서 소개했던 조합 개념을 이용해서 접근했다.

  • 3개를 뽑아서 더해야 하니까 반복문은 총 3개를 사용했다.
    -> 2개 뽑는 경우 반복문은 아래와 같이 작성을 했는데

  for i in range(len(numbers)-1):
        for j in range(i+1,len(numbers)):

  3개를 뽑는 경우니까 아래와 같이 작성 했다. (반복문의 안쪽으로 갈 수록 시작점은 그 바로 바깥쪽 반복문 시작점보다 1 큰 값 부터 시작, 종료지점은 제일 안쪽 바깥문이 list의 length까지 돌리고 바깥의 반복문은 1씩 작아지는 원리)

  for i in range(len(a)-2):
       for j in range(i+1,len(a)-1):
           for k in range(j+1, len(a)):

  • 이렇게 해서 인덱스 i,j,k에 해당하는 배열의 원소의 값을 sum 변수에 저장

    sum = a[i]+a[j]+a[k]

  • sum 값에 대해서 소수인지 아닌지 판별 해야 하기 때문에 아래와 같이 반복문 설정 (1부터 sum값 까지 1씩 증가시킨 값으로 % 연산의 결과가 0인 경우엔 약수로 볼 수 있으니)

    for num in range(1, sum+1):
           if sum % num == 0:
                yaksu.append(num) #약수인 경우 yaksu란 list에 append

  • 그렇게 sum 값에 대해서 약수를 구한 list인 yaksu의 길이가 2인 경우엔 1과 자기자신만 포함한 경우 이므로 answer 변수의 값을 1 증가시킴

    if len(yaksu) == 2:
         answer += 1

  • 그리고 해당 sum 값에 대해서 반복문이 끝나고 나면 변수 sum과 list인 yaksu 를 초기화 한다.

    sum = 0
    yaksu = []

  • 아래는 최종 코드

    def solution(a):
    answer = 0
    sum = 0
    yaksu = []
    
    for i in range(len(a)-2):
        for j in range(i+1,len(a)-1):
            for k in range(j+1, len(a)):
                sum = a[i]+a[j]+a[k]
                for num in range(1, sum+1):
                    if sum % num == 0:
                        yaksu.append(num)
                if len(yaksu) == 2:
                    answer += 1
                sum = 0
                yaksu = []
                    
    return answer

아쉬운 점

• 1은 모든 수의 약수이기 때문에 1부터가 아닌 2부터로 잡아도 된다.
  • 그 경우엔 yaksu list의 길이가 2가 아닌 1로 잡으면 됨.
• 또한 for num in range(1,sum+1) 에서 sum+1까지가 아닌 sum 까지로 계산
  • yaksu list의 길이가 0으로 잡으면 동일한 알고리즘.