[Programmers 코딩테스트 연습_Lv1_Python] 소수만들기

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

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

 

[ 문제 ]

 

[ 나의 귀여운 풀이 ]

from itertools import combinations

def eratosthenes_sieve(size):
    aristo = [True] * size
    aristo[0] = False
    aristo[1] = False
    end = int(size ** 0.5)

    for i in range(2, end+1):
        if aristo[i]:
            for j in range(i + i, size, i):
                aristo[j] = False

    return aristo

def solution(nums):
    prime = aristo_sieve(3001)
    combi = list(combinations(nums, 3))
    answer = 0
    for i in range(len(combi)):
        if prime[sum(combi[i])]:
            answer +=1
    return answer

 

[ 배운 것 ]

 

1. 에라토스테네스의 체(Eratosthenes Seive): https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes

 

[ 개인적인 만족도 ]

(수달 1~5마리)

수달: 3마리!

 

꾸준히 열심히!