[1978]소수 찾기

import sys

N = int(sys.stdin.readline())

nums = list(map(int, sys.stdin.readline().split()))

count = 0

for num in nums:
    if num == 1:
        continue
    elif num ==2:
        count += 1
    else:
        for i in range(2, num):
            if num % i == 0:
                count -= 1
                break
        count += 1

print(count)

에라토스테네스의 체 => 참고하여 다시 풀어보기

1978 소수 찾기