본문 바로가기
Beakjoon&프로그래머스/파이썬

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

by 현장 2021. 12. 11.

-코드

def plus(a, b, c):
    r = a + b + c
    for i in range(2, r):
        if r % i == 0:
            return False
    return True

def solution(nums):
    answer = 0
    for i in range(len(nums) - 2):
        for j in range(i + 1, len(nums) - 1):
            for k in range(j + 1, len(nums)):
                if plus(nums[i], nums[j], nums[k]):
                    answer += 1
    return answer

처음에 하나의 함수에 다 넣으려고 하니 너무 길어져서 찾아보니 나눠도 되는 것을 알고 나눠서 작성을 했습니다.