[알고리즘]백준 4134:

by

문제

정수 n(0 ≤ n ≤ 4*109)이 주어지면 n보다 크거나 같은 가장 작은 소수를 찾는 프로그램을 작성하세요.

기입

테스트 사례의 수는 첫 번째 줄에 지정됩니다. 각 테스트 케이스는 한 줄로 구성되며 정수 n이 주어집니다.

누르다

각 테스트 케이스에 대해 n보다 크거나 같은 가장 작은 소수를 한 줄에 하나씩 인쇄합니다.

import sys 
import math
N = int(sys.stdin.readline().strip())

def is_prime(n): 
    for i in range(2,int(math.sqrt(n))+1 ):
        if n%i ==0:
            return False
    return True 

for i in range(N):
    num = int(sys.stdin.readline().strip())
    if num ==0 or num ==1:
        print(2)
    else: 
        while True:
            if is_prime(num)==True:
                print(num)
                break 
            num+=1