[Python] 백준 1978 - 소수 찾기

문제

주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.

 

입력

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

 

출력

주어진 수들 중 소수의 개수를 출력한다.

---

정답

import math
def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, int(math.sqrt(num)+1)):
        if num % i == 0:
            return False
    return True

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

count = 0

for x in range(n):
    if is_prime(a[x]) == True:
        count += 1

print(count) 

소수를 구하고, 소수면 count++ 해준다.