풀이_소수만들기.md

🥳 프로그래머스 소수 만들기

• Date : 2021.01.27(수)
• Time : 10분

문제

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

• 제한사항

  • nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
  • nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.
    
    

코드 풀이

from itertools import combinations

    answer = 0
    for a in combinations(nums, 3):
        if is_prime(sum(a)):
            answer += 1
    return answer

: 먼저 combinations는 순열을 구해주는 itertools 모듈의 원소입니다. combinations를 이용해 원소개 3개인 조합을 구해서 for문으로 돌려줍니다. 그리고 소수인지 확인하는 함수를 이용해 소수이면 answer에 +1를 해줍니다.

def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, num):
        if num % i == 0:
            return False
    return True

: 소수인지 판단해주는 함수를 만듭니다. 숫자가 2 이상일때만 소수를 판단할 수 있습니다. 만약에 소수가 2,num-1 안에서 아무 숫자로도 나눠지지 않는다면 그것은 1과 자기자신으로만 나눠지는 소수의 조건에 부합함으로 소수라고 판단할 수 있습니다. 그렇다면 True를 return 하고 한번이라도 다른 수로 나눠진다면 바로 False를 return 해줍니다.