백준) 1978 - 소수 찾기

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

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

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

N = int(input())
n_count = 0
for num in map(int, input().split()):
  check = 0
  for i in range(1, num+1):
    if num % i == 0: check += 1
  if check == 2: n_count += 1
print(n_count)

풀이 : 띄어쓰기를 통해 구분되는 숫자들을 입력받은 뒤 각 숫자(num)마다 1부터 num까지의 숫자로 나누어보며 나머지를 계산하여 소수인지 아닌지를 판별한다. 만약 num이 소수라면 n_count에 1을 더해준다.