소수 만들기

이건 진짜 생각이 안나서 참고 했다.

from itertools import combinations 이 걸 배워야 겠다 한번 정리 한다.

from itertools import combinations          # 중복허용(X), 순서의미(O) 인 조합을 import

"""소수 판별 함수"""
def is_prime_number(num):
    if num==0 or num==1:
        return False
    else:
        for n in range(2, (num//2)+1):      # math를 사용하지 않고 (num//2)+1 까지로 설정
            if num%n == 0:
                return False
        
        return True

def solution(nums):
    answer = 0
    cmb = list(combinations(nums,3))        # nums배열을 3개씩 조합 후 list로 변경
    for arr in cmb:
        if is_prime_number(sum(arr)):       # cmb를 하나씩 가져와 sum한 값을 소수 판별 함수로 넘겨줌
            answer += 1                     # return 값이 True이면 count(=answer) +1
    
    return answer

 

velog.io/@insutance/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4Python-%EC%86%8C%EC%88%98-%EB%A7%8C%EB%93%A4%EA%B8%B0

[프로그래머스][Python] 소수 만들기