Summer/Winter Coding(~2018) > 소수 만들기 - Pyton, Combinations, 조합, 소수

 

 

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

프로그래머스

 

 

 

풀이 - 나의 생각

itertools.combinations 를 사용해서 조합을 하고

 

그 조합의 합을 소수인지 판별하면 됩니다.

 

 

 

코드

 

from itertools import combinations

def is_prime(num):
    for i in range(2, int(num**0.5) + 1):
        if num%i == 0:
            return False
    return True

def solution(nums):
    answer = 0
    for comb in combinations(nums, 3):
        if is_prime(sum(comb)):
            answer += 1
    return answer

 

 

 

 

 

 

 

다음에 또 봐요