Level 1. 소수 만들기

03. 소수 만들기

코딩테스트 연습 > Summer,Winter Coding(~2018) > 소수 만들기
https://programmers.co.kr/learn/courses/30/lessons/12977

문제 설명

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

제한사항

ㆍ nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.

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

입출력 예

nums = (1,2,3,4) result = 1

구상

1. itertools의 combination 라이브러리를 이용하여 주어진 숫자 중 3개를 더한 수를 가지를 리스트를 만든다.

2. 해당 리스트의 원소(리스트)의 합을 2부터 원소값까지 나누어 봤을 때,

   2-1. 나누어 떨어지는 경우가 발생할 경우 다음 원소로 넘어간다.

   2-2. 나누어 떨어지지 않을 경우 result에 +1을 한다.

풀이 방법

구상한 방법을 이용한 풀이

from itertools import combinations
def solution(nums):
    count = 0
    sums = itertools.combinations(nums,3)

    for i in sums:
        num = sum(i)
        for j in range(2,num):
            if num % j == 0:
                break
        else:
            count +=1
    return count

에라토스테네스의 체 방법

from itertools import combinations
def prime_number(x):
    answer = 0
    for i in range(1,int(x**0.5)+1):
        if x%i==0:
            answer+=1
    return 1 if answer==1 else 0

def solution(nums):
    return sum([prime_number(sum(c)) for c in combinations(nums,3)])