[백준] 다음 소수

0과 1만 예외처리 해주면 되는 문제였다.
isPrime을 통하여 입력받는 인자의 제곱근까지 확인하며 나누어지는 경우 False를 반환하고, 끝까지 처리된 경우 True를 반환하면 된다.

def isPrime(K):
    
    if K == 0 or K == 1:
        return False
    
    for i in range(2,int(sqrt(K))+1):
        if K % i == 0:
            return False
        
    return True


N = int(input())
for i in range(N):
    a = int(input())
    
    while True:
        if isPrime(a):
            print(a)
            break
            
        a += 1