[프로그래머스-Level 1] 소수 만들기_파이썬

programmers.co.kr/learn/courses/30/lessons/12977

코딩테스트 연습 - 소수 만들기

---

from itertools import combinations


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

def solution(nums):
    count = 0
    for nums_case in combinations(nums, 3):
        if is_prime(nums_case):
            count += 1
    
    return count

조합과 소수 판별 알고리즘이 합쳐진 문제이다.

itertools 를 사용할 수 있고 소수 판별 알고리즘을 알고 있다면 쉽게 풀 수 있다.