[프로그래머스] 소수찾기 - 파이썬 - 에라토스테네스의 체

 

문제 https://programmers.co.kr/learn/courses/30/lessons/12921

코딩테스트 연습 - 소수 찾기

 

import math 
def solution(n):
    arr = [True for i in range(n+1)]

    for i in range(2,int(math.sqrt(n))+1):
        if arr[i] == True:    
            j = 2
            while i * j <=n:
                arr[i*j] = False
                j+=1
    answer = len([i for i in range(2,n+1) if arr[i] == True])
    return answer
print(solution(5))

참고: https://wikidocs.net/21638