1978 - 소수 찾기

문제

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

입력

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

출력

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

나의 정답

n = int(input())

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

total_count = 0 # 총 소수의 개수

for value in array:
    count = 0
    # value 사이의 모든 수로 나눠 소수인지 판단
    for i in range(1, value+1):
        if value % i == 0:
            count += 1
        if count > 2:
            break # 소수 탈락 시 break
    if count == 2: # 소수라면 개수 up
        total_count += 1

print(total_count)