본문 바로가기
알고리즘/프로그래머스 level 1,2

[파이썬🐍] 프로그래머스 : 소수 만들기

by 코딩개미뚠뚠 2021. 4. 20.
반응형
from itertools import combinations
def check(a,b,c):
    sum = a+b+c
    for i in range(2,sum):
        if sum%i == 0 : return False
    return True
def solution(nums):
    answer = 0
    A = list(combinations(nums,3))
    for i in A :
        if check(i[0],i[1],i[2]) : answer += 1
    return answer

"check"라는 소수인지 확인해주는 함수를 추가해줬다.

여기서도 combinations을 활용할 수 있었다.

한번 기억하니 아주 유용하다ㅎㅎㅎ

반응형

댓글