[프로그래머스] 소수 찾기 - python

소수 찾기

문제 설명

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.

소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다. (1은 소수가 아닙니다.)

제한사항

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

풀이

아리스토테네스의 체 구현

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

다른 사람 풀이

아리스토테네스의 체를 더 간편하게 구현한 버전

range와 set로 범위내 집합을 만든 후 전체에서 빼버린다.

def solution(n):
    num=set(range(2, n + 1))
​
    for i in range(2, n + 1):
        if i in num:
            num-=set(range(2 * i, n + 1, i))
                
    return len(num)