[백준] 소수 찾기

소수 찾기

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

나의 풀이

소수 판별 함수를 따로 만들어서 확인

from math import sqrt, floor
n = int(input())
arr = list(map(int,input().split()))
total = 0

def is_prime(n):
    if n == 1:
        return False
    div = floor(sqrt(n))
    for i in range(2,div +1):
        if n % i == 0:
            return False
    return True

for i in arr:
    if is_prime(i):
        total += 1
print(total)