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

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

from itertools import combinations

def is_prime(n): # 소수 판별
    for i in range(2, int(n ** (1/2)) + 1):
        if n % i == 0:
            return False
    return True

def solution(nums):
    answer = 0
    
    for combi in combinations(nums, 3): # 3개의 숫자로 이루어진 조합 만들기
        plus = sum(combi)
        
        if is_prime(plus): # 합이 소수면 +1
            answer += 1

    return answer