[백준] #1978 - 소수 찾기 (파이썬, Python)

소수 찾기

https://www.acmicpc.net/problem/1978

내가 쓴 코드

에라토스테네스의 체를 이용하면 소수를 쉽고 빠르게 구할 수 있다.

n = int(input())
a = list(map(int, input().split()))

prime_nums = [1] * 1001
prime_nums[1] = 0
for i in range(2, 1001):
    if prime_nums[i]:
        for j in range(i + i, 1001, i):
            prime_nums[j] = 0

ans = 0
for i in a:
    ans += prime_nums[i]

print(ans)

좋은 웹페이지 즐겨찾기