[Python] 백준 1978번(소수 찾기) 문제 풀이

 

def is_prime(n):
    if n == 1:
        return False
    if n % 2 == 0:
        if n == 2:
            return True
        return False
    if n % 3 == 0:
        if n == 3:
            return True
        return False
    if n % 5 == 0:
        if n == 5:
            return True
        return False
    if n % 7 == 0:
        if n == 7:
            return True
        return False
    if n % 11 == 0:
        if n == 11:
            return True
        return False
    if n % 13 == 0:
        if n == 13:
            return True
        return False
    if n % 17 == 0:
        if n == 17:
            return True
        return False
    if n % 19 == 0:
        if n == 19:
            return True
        return False
    if n % 23 == 0:
        if n == 23:
            return True
        return False
    if n % 29 == 0:
        if n == 29:
            return True
        return False
    for i in range(30, n):
        if n % i == 0:
            return False
    return True
    
answer = int(input())
hihi = list(map(int, input().split()))
for n in hihi:
    if is_prime(n) == False:
        answer -= 1
        
print(answer)

 

 내가 소수를 어디까지 알고 있나 궁금해서 이런 뻘짓을 한번 해봤다.