[알고리즘 문제 풀이][파이썬] 백준 1978번: 소수 찾기

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

📑 문제 설명

주어진 수 N개 중에 소수가 몇개인지 출력하는 프로그램 작성

입력: 수의 개수, 주어진 개수만큼 숫자
출력: 주어진 숫자 중 소수의 개수

💡 문제 해결 방법

소수는 1과 자기 자신을 제외하고 약수가 없는 수를 의미한다.
따라서 1과 자기 자신을 제외한 모든 수로 나눈 후 나머지가 0이 되지 않는 숫자를 찾아 count 해주면 된다.

💻 코드

import sys


def pn(num):
    if (num == 1):
        return 0
    for i in range(2, num):
        if (num%i==0):
            return 0
    return 1


if __name__ == '__main__':
    t = int(sys.stdin.readline())
    n = sys.stdin.readline().split()
    num_list = list()
    cnt = 0
    for i in range(t):
        cnt += pn(int((n[i])))
    print(cnt)

💟 추가적으로 알게 된 점