[파이썬] 소수 구하기

import math

## Parameter
# lst : 0~N까지의 자연수를 담은 리스트
def problem(lst):
    N = len(lst)-1
    for i in range(2, math.floor(math.sqrt(N))+1):
        prime = True
        for j in range(2, math.floor(math.sqrt(i))+1):
            if i!=j and i%j==0:
                prime=False
                break
        if prime:
            for j in range(2, N//i+1):
                lst[i*j] = 0
        
    return lst

'에라토스테네스의 체'를 사용한 알고리즘