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

0. 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12977

프로그래머스

 

1. 문제 설명

1) 주어진 배열에서 3개의 숫자를 더한 후

2) 소수인지 판단

3) 주어진 배열에서 가능한 조합의 개수 구하기

 

2. 입출력

# input
nums = [1, 2, 7, 6, 4]

# output
answer = 4

 

3. 코드

def isPrime(temp):
    if temp > 1:
        for i in range(2, temp):
            if temp % i == 0:
                return False
    else:
        return False
    return True

def solution(nums):
    # nums : 주어진 숫자의 배열
    answer = 0
    len_num = len(nums)

    for i in range(len_num):
        for j in range(i+1, len_num):
            for k in range(j+1, len_num):
                temp = nums[i] + nums[j] + nums[k]
                if isPrime(temp) == True:
                    answer += 1
    return answer

 

실행 결과

 

4. 알게된 점

1) boolean을 사용하여 판별하여 간단하게 코드 구현

 

5. Github

https://github.com/iihye/Algorithm/blob/main/Programmers/make_prime.py

GitHub - iihye/Algorithm