소수 구하기_Python

Rocki_log·2022년 8월 6일
0

Algorithm

목록 보기
3/3
post-thumbnail

소수 리스트 구하기

  • 에라토스테네스의 체
def solution(n):
    if n < 2:
        return []
    s = [0, 0] + [1] * (n - 1)
    for i in range(2, int(n**.5)+1): #데칼코마니이기에 반만 구해도 된다.
        if s[i]:
            s[i*2::i] = [0] * ((n - i)//i)
    answer = [i for i, v in enumerate(s) if v]
    return answer
profile
Learning&Running

0개의 댓글