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

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

---

● 문제

• 주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다.
• 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

● 소스코드

from itertools import combinations

def solution(nums):
    answer = 0
    
    candidate = list(combinations(nums,3))
    
    for i in range(len(candidate)):
        cur = sum(candidate[i])
        is_prime = True
        for j in range(2,cur):
            if cur % j == 0:
                is_prime = False
                break
        if is_prime == True:
            answer += 1

    return answer

● 풀이

• 특별한 알고리즘은 없다.
• itertools의 combinations 함수를 사용하여 nums list에서 숫자 3개로 이루어진 모든 조합을 얻을 수 있다.
  • "combinations", 리스트 원소들의 조합 구하기

[Python] "combinations", 리스트 원소들의 조합 구하기

• 각 경우의 수에 대해 3개의 숫자의 합이 소수인지 판별해 주면 된다.
  • 소수는 1과 자기 자신 외의 다른 수로 나누어 떨어지지 않는다.

 

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

코딩테스트 연습 - 소수 만들기