[백준 - python] 1978번: 소수 찾기

<문제 링크>

1978번: 소수 찾기

 

<문제 풀이>

N = int(input())
num_list = map(int, input().split())
prime_num = []

for num in num_list:
  if num == 1:
          continue
  for i in range(2, int(num ** 0.5) + 1 ):
      if num % i==0:
          break
  else:
    prime_num.append(num)

print(len(prime_num))

1978 입출력 예시

 

*key point: 2 ~ 제곱근까지만 나누어 보는 것만으로도 소수임을 판별 가능하다. for - else문의 활용으로 표현을 간결히 한다.