[python] 백준 4134 다음 소수
문제
풀이
import sys
import math
input = sys.stdin.readline
n = int(input())
lst = []
for i in range(n):
lst.append(int(input())) #입력값 리스트
def is_prime_number(x): #소수 판별 함수
for i in range(2,int(math.sqrt(x))+1):
if x % i == 0:
return False
return True
for i in lst:
x = i
if x == 0 or x == 1: #degenerate case x가 0이거나 1일 경우
print(2)
continue
else:
while is_prime_number(x) == False: #x가 소수가 될 때까지 x += 1
x += 1
print(x) #가장 가까운 소수 출력