1978번 - 소수 찾기

문제

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

입력

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

출력

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

나의 제출 - Python

소수 찾기 문제를 풀 때는 유명한 에라토스테네스의 체를 사용했다. 2를 제외한 2의 배수, 3을 제외한 3의 배수를 False로 바꿔주면서 소수만 True로 남긴다.

 

N = int(input())
listinin = [int(x) for x in input().split()]

prime = [False,False] + [True]*(1000-1)
for i in range(2,len(prime)):
    if prime[i] == True:
        for j in range(i*2,len(prime),i):
            prime[j] = False
    
printprime = [i for i in listinin if prime[i]==True]
[print(len(printprime))]