[Level 2] 소수 만들기

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

프로그래머스

▶ 코드

# 브루트 포스 -> itertools의 '조합'(=순서가 없는 조합) 사용

from itertools import combinations

def isPrime(n):
    for i in range(2, n):
        if n % i == 0:
            return False
    return True

def solution(nums):
    cb = combinations(nums, 3) # 주어진 숫자 중 3개 뽑은 조합  
    #print(list(c))

    c = 0
    for a in list(cb):
        if isPrime(sum(a)):
            c += 1
    #print(c)
    return c

#solution([1,2,3,4])
#solution([1,2,7,6,4])