[BOJ / Python] 1978 소수 찾기

문제

백준 1978 소수 찾기

코드

#BOJ 1978 약수 찾기

n = int(input())
nums = list(map(int, input().split()))
count = 0
for x in nums:
    error = 0
    if x > 1:
        for i in range(2, x):
            if x % i == 0:
                error += 1
        if error == 0:
            count += 1

print(count)

코드 설명
소수의 개수를 셀 변수 count를 선언하고 0으로 초기화한다.
소수가 아님을 판단하기 위해 사용할 변수 error을 선언하고 0으로 초기화한다. 해당 변수는 입력값이 새로 들어올 때마다 초기화되어야하므로 for문 안에서 초기화되어야 한다.

1은 소수가 아니므로 입력값이 1보다 큰 경우에 대해 다룬다. 입력값이 n일 때, 2부터 n-1까지의 수로 나누었을 때 나누어떨어지면 error 변수의 값을 올린다.
for i in range(2, x) 해당 for문이 종료되면 error 변수의 값이 0인지 확인하고 0인 경우에만 소수이므로 count 값을 올린다.