[백준 - python] 4134번: 다음 소수

문제 링크

4134번: 다음 소수

문제 풀이

from sys import stdin

input = stdin.readline

test_case = int(input())

for i in range(test_case):
    n = int(input())

    while 1:
        if n == 0 or n == 1:
            print(2)
            break

        for i in range(2, int(n ** 0.5) + 1 ):
            if n % i == 0:
                break
        else:
            print(n)
            break
        n += 1

4134 입출력 예시

*key point: 2부터 n의 제곱근까지의 수들로 나누어보며 약수가 있는지 확인한다. n=0, n=1 인 경우를 따로 고려해주어야 한다.