프로그래머스. 소수 만들기 파이썬 풀이

프로그래머스. Summer/Winter Coding(~2018). Level 1. 소수 만들기 파이썬 풀이

문제링크 https://programmers.co.kr/learn/courses/30/lessons/12977

combinations 라이브러리를 사용한다면 쉬운 문제
combinations 라이브러리 설명 https://velog.io/@minan/파이썬-문법-정리-for-코딩-테스트

from itertools import combinations

# 소수 판별 알고리즘
def is_prime_number(x):
    for i in range(2, x):
        if x % i == 0:
            return False
    return True
    

def solution(nums):
    answer = 0
    
    # 3개를 뽑는 모든 경우의 수
    array = list(combinations(nums, 3)) 
    
    for num in array:
        if is_prime_number(sum(num)) == True:
            answer += 1 

    return answer