프로그래머스 소수 만들기 파이썬

문제링크

import itertools

# 소수를 찾는 알고리즘
def prime_num(n):
    for i in range(2, n):
        if n % i == 0:
            return False
    return True

def solution(nums):
    answer = 0

	# itertools.combinations(nums, 3)은 3가지의 조합을 return 해 준다.
    for i in itertools.combinations(nums, 3):
        # 조합의 모든 합을 소수 알고리즘에 넣어서
        cur_sum = sum(i)
        # 소수이면 답에 +1을 해 준다.
        if prime_num(cur_sum):
            answer += 1
        
    return answer