[BaekJoon] 4134번 : 다음 소수 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 다음 소수 입니다.

4134번: 다음 소수

👨🏻‍💻 코드 ( Solution )

import math 


def is_prime_number(number):
    is_prime = True
    
    if number == 0 or number == 1:
        is_prime = False
    else:
        for num in range(2, int(math.sqrt(number))+1):
            if number % num == 0:
                is_prime = False
                break
                
    return is_prime



def next_primary_number(number): 
    while True:
        is_prime = is_prime_number(number=number)
        
        if is_prime:
            answer = number
            break
        number += 1
            
    return answer
            

if __name__ == "__main__":
    for _ in range(int(input())):
        number = int(input())
        print(next_primary_number(number=number))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07