[프로그래머스] Summer/Winter Coding(~2018) > 소수 만들기

combinations 에 대해서 확실하게 사용할 수 있도록 하자.

 

 

from itertools import combinations

def isPrime(n):
    if n==1 or n==0 :
        return False
    elif n>1:
        for i in range(2,round(n/2)):
            if n%i == 0:
                return False 
        return True
        
def solution(nums):
    answer = 0
    
    combin = list(combinations(nums,3))
    for arr in combin:
        if isPrime(sum(arr)):       
            answer += 1
    
    return answer