[백준] / [Python] / [1978] 소수 찾기

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

1978번: 소수 찾기

---

문제

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

입력

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

출력

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

---

import sys
input = sys.stdin.readline

def isPrime(n):
    if n == 1:
        return -1
    elif n == 2:
        return 1
    
    for i in range(2, n // 2 + 1):
        print(i)
        if n % i == 0:
            return -1
    else:
        return 1

N = int(input())
nums = list(map(int, input().split()))
    
s = 0
for n in nums:
    if isPrime(n) == 1:
        s += 1
        
print(s)