[백준/Python] 1978번 소수 찾기 문제

■ 1978번 소수 찾기 문제

 

출처: 백준 1978번 소수 찾기 문제

 

 

■ 코드 풀이

 

먼저 1개의 숫자가 소수인지를 판단하는 is_it_sosu 함수를 선언합니다. 1은 소수가 아니고 자기 자신의 절반까지만 확인하면 소수인지를 알 수 있으므로 이를 참고하여 range를 설정합니다. 이 함수는 숫자가 소수이면 True(1)를, 소수가 아니면 False(0)을 반환합니다.

숫자를 list로 입력받고, 개별 원소 별로 is_it_sosu 함수를 호출합니다. 숫자 1은 소수가 아니므로, 스킵하고 2부터 count 하면 됩니다.

def is_it_sosu(num):
    sosu = True
    
    for i in range(2, num//2+1):
        if num % i == 0:
            sosu = False
            break
            
    return sosu

        
N = int(input())
num_list = list(map(int, input().split()))

count = 0

for num in num_list:
    if num == 1:
        continue
    count += is_it_sosu(num)

print(count)