1978. 소수 찾기

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

 

def is_prime(n):
    if n == 1:
        return False
    else:
        div_cnt = 0
        for i in range(2,int(n**0.5)+1):
            if n % i == 0:
                div_cnt += 1
                break
        if div_cnt == 0:
            return True
        else:
            return False

N = int(input())
nums = list(map(int, input().split()))
prime_cnt = 0

for num in nums:
    if is_prime(num):
        prime_cnt += 1
 
print(prime_cnt)

 

 

• 먼저 어떤 숫자가 소수인지 아닌지를 판별하는 is_prime 함수를 만들어주었습니다. 
  소수 여부를 판단하는 방법은 2부터 n의 제곱근까지의 숫자 중 n을 나누어 떨어지게 할 수 있는 숫자가 존재하면 소수가 아니고, 존재하지 않으면 소수입니다.
  
  
• 만든 is_prime 함수를 이용해서 입력 받은 숫자들 중 소수의 갯수를 세줍니다.