BOJ/백준-1978-python

문제📖

풀이🙏

• 첫 줄에 수의 개수 N이 주어진다.
• 다음으로 N개의 수가 주어지는데 수는 1000 이하의 자연수이다.
• 주어진 수들 중 소수의 개수를 출력한다.
  -> sum + for 반복문 + if 조건문을 한번에 함수형 프로그래밍으로 선언하여 구현했다.

코드💻

# boj, 1978 : 소수 찾기, PyPy3
import sys

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

def get_result():
    num_list = list(map(int, sys.stdin.readline().split(' ')))
    
    return sum(1 for i in num_list if check_sosu(i))

N = int(sys.stdin.readline())

print(get_result())

결과😎

출처 && 깃허브📝

https://www.acmicpc.net/problem/1978
github