[알고리즘] 범위내의 소수 찾기

코드

def find_primes(n):
    prime_nums = [True] * (n+1)
    prime_nums[0] = prime_nums[1] = False

    for i in range(2, int(n ** 0.5) + 1):
        if prime_nums[i]:
            for j in range(i*i, n+1, i):
                print(f"   j = {j}")
                prime_nums[j] = False
    
    return [i for i, is_prime in enumerate(prime_nums) if is_prime]

print(find_primes(int(input())))

입력값

30

출력값

[2, 3, 5, 7, 11, 13, 17, 19, 23, 29]