[알고리즘] 소수 판별

코드

def is_prime(n):
    if n == 2:				# 2는 소수
        return 1
    if n < 2 or n%2 == 0:	# 0, 1, 짝수는 소수가 아님
        return 0
    for i in range(2, int(n**0.5)+1): # 루트 값 이하에 약수가 없다면 소수 (1 제외)
        if n % i == 0:
            return 0
    return 1

input() # 첫번째 input 안 씀
nums = list(map(int, input().split()))
print(sum([is_prime(i) for i in nums]))

입력값

4
1 3 5 7

출력값

3

입력값

4
1 2 3 5

출력값

3

---

참고문제

백준 1978번: 소수 찾기 (브론즈2)