백준(BOJ) - 소수 찾기(1978번) - 파이썬(python)

 소수판별 함수 참고: https://jinhan0716.tistory.com/5

소수(prime number) 판별 코드 - 파이썬(python)

import sys
input=sys.stdin.readline

def is_prime(n):
    if n<2:
        return False
    if n in (2,3):
        return True
    if n%2==0 or n%3==0:
        return False
    if n<9:
        return True
    k,l=5,n**0.5
    while k<=l:
        if n%k==0 or n%(k+2)==0:
            return False
        k+=6
    return True

n=int(input().rstrip())

arr=list(map(int,input().split()))

cnt=0
for x in arr:
    if is_prime(x):
        cnt+=1
print(cnt)