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

문제

문제 링크

코드

from itertools import combinations #조합
import math

def is_prime(n):
    for i in range(2,int(math.sqrt(n)) + 1): #제곱근까지
        if n%i==0:
            return True
            
    return  False   



def solution(nums):
    answer = 0  
    com=list(map(lambda x:sum(x),list(combinations(nums,3)))) 
    #3개 조합된 수의 합을 바로 리스트로
    for i in com:
        if is_prime_number(i) is False: #소수인 경우
            answer+=1

    return answer

출처: 프로그래머스