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

문제

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

입력

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

출력

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

---

풀이

import sys
input = sys.stdin.readline

n = int(input())
num = list(map(int, input().split()))
count = 0
answer = 0

for i in num:
    for j in range(1, i+1):
        if(i == 1):
            break
        elif(i%j == 0):
            count += 1
    if(count == 2):
        answer += 1
        count = 0
    else:
        count = 0
print(answer)

 

문제 출처

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

1978번: 소수 찾기