[Level 1 / Python] 소수 찾기

문제 설명

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요. 소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다. (1은 소수가 아닙니다.)

• n은 2이상 1000000이하의 자연수입니다.

출처 : https://programmers.co.kr/learn/courses/30/lessons/12921

풀이 방법

소수를 찾기 위해 에라토스테네스의 체를 사용하여 풀었습니다. 에라토스테네스의 체는 고대 그리스 수학자 에라토스테네스가 발견하였습니다.

소스 코드

def solution(n):
    nums = [False] + [True] * n
    for i in range(2, int(n**0.5)+1):
        if nums[i] == True:
            for j in range(2*i, n+1, i):
                nums[j] = False
    answer = [k for k in range(2, n+1) if nums[k] == True]
    return len(answer)