[Python] 백준 4134 다음 소수

백준 4134 다음 소수
문제

정수 n(0 ≤ n ≤ 4*10^9)가 주어졌을 때, n보다 크거나 같은 소수 중 가장 작은 소수 찾는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다.

출력

각각의 테스트 케이스에 대해서 n보다 크거나 같은 소수 중 가장 작은 소수를 한 줄에 하나씩 출력한다.

---

제출

import sys
input=sys.stdin.readline
n=int(input())
def ch(x):
    for i in range(2,int(x**0.5)+1):
        if x%i==0:
            return False
    return True
for _ in range(n):
    i=int(input())
    while True:
        if i==0 or i==1:
            print(2)
            break
        if ch(i):
            print(i)
            break
        else:
            i+=1

예제

3
6
20
100

결과

백준 4134 다음 소수

 

4134번: 다음 소수