[Python] 프로그래머스 소수 만들기

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 

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

 

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

프로그래머스

제한 사항

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

예제

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

문제 풀이

# itertools 사용
import itertools

def solution(nums):
    answer = 0
    
    # 서로 다른 3개의 수 조합 만들기
    comb = list(itertools.combinations(nums, 3))
    
    for idx in comb:
        tmp = sum(idx)
        # 소수 판단 플래그 변수
        is_prime = True
        
        # 소수 판별
        for i in range(2, int(tmp**0.5)+1):
            if tmp % i == 0:
                is_prime = False
                break
                
        if is_prime:
            answer += 1
        
    return answer

# itertools 미사용
def solution(nums):
    answer = 0
    
    # 서로 다른 3개의 수 조합 만들기
    for i in range(len(nums)):
        for j in range(i + 1, len(nums)):
            for k in range(j + 1, len(nums)):
                tmp = nums[i] + nums[j] + nums[k]
                # 소수 판단 플래그 변수
                is_prime = True
                
                # 소수 판별
                for i in range(2, int(tmp**0.5) + 1):
                    if tmp % i == 0:
                        is_prime = False
                        break

                if is_prime:
                    answer += 1

    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.