Baekjoon Online Judge 1978번 파이썬

문제 링크
https://www.acmicpc.net/problem/1978

1978번: 소수 찾기

def prime_number(number):
  if number != 1:
    for i in range(2,number):
      if number % i == 0:
        return False
  else:
    return False
  return True

이 함수를 먼저 작성해서 문제에 적용시켰다.
먼저 1이 아닌 숫자를 1과 자기 자신으로 나누어지면 False를 반환시켜서 한번 거르고,
else를 이용하여 1을 걸렀다.
마지막에 return True 를 이용하여 두 조건에 해당되지 않는 수를 True로 반환시켰다.


정답 코드

import sys

N = int(sys.stdin.readline())
numbers = list(map(int,sys.stdin.readline().split()))

def prime_number(number):
  if number != 1:
    for i in range(2,number):
      if number % i == 0:
        return False
  else:
    return False
  return True
  
result = []
for i in range(N):
  if prime_number(numbers[i]) == True:
    result.append(numbers[i])
  
print(len(result))