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

from itertools import combinations

def solution(nums):
    global answer
    answer = 0

    def is_prime(num):
        global answer

        cnt = 0
        for i in range(1, num+1):
            if num % i == 0:
                cnt += 1
        if cnt == 2:
            answer += 1
            return
        
    prime_lst = list(combinations(nums, 3))
    
    for i in prime_lst:
        is_prime(sum(i))
        
    return answer

combinations 모듈을 통해 해당 배열의 경우의 수를 구한 후 소수를 판별하는 함수에 넣어서 풀이.