[파이썬]프로그래머스 Lv.1 소수만들기

✅ 코드

from itertools import combinations
import math

def is_prime(x):
    for i in range(2, int(math.sqrt(x)) + 1):
        if x % i == 0:
            return False
    return True

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

주어진 배열에서 3개의 수를 조합해 더하면 소수가 되는 케이스의 수를 구해야 합니다.

다행스럽게도(?) 파이썬의 itertools 모듈을 사용하면 수들의 조합을 구할 수 있습니다.

소수 여부를 판별하는 is_prime 함수를 별도로 선언해 문제를 해결했습니다.