백준 1978 : 소수 찾기 (파이썬)

https://www.acmicpc.net/problem/1978 문제풀이

1978번: 소수 찾기

---

문제

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

입력

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

출력

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

예제 입력 1

4
1 3 5 7

예제 출력 1

3

---

코드

import sys
sys.stdin.readline()
nums = list(map(int,sys.stdin.readline().split()))
answer = 0
for num in nums :
    if num == 1 : continue
    for i in range(2,int(num**.5)+1) :
        if num%i ==0 :
            break
    else : answer +=1
print(answer)

코드 해설

간단하게 각 숫자마다 소수인지 판별하는데 1은 소수가 아니니 바로 넘기고 소수 판별 알고리즘을 돌리면 된다.

여기서 중요한 점은 소수판별을 할 때 2부터 제곱근까지만 확인을 하는 것이다.

제곱근까지만 확인을 해도 해당 숫자가 소수인지 확인할 수 있기 때문에 최대한 for문을 줄여 효율성을 높인 것!!!!