[파이썬/python] 백준 4134번 : 다음 소수(🥈4)

https://www.acmicpc.net/problem/4134

4134번: 다음 소수

#소수를 판별하는 함수
def prime(x):
    if x == 0 or x == 1:
        return False
    else:
        for i in range(2,int(n**0.5)+1):
            if x % i == 0 :
                return False #소수가 아님
        return True #소수가 맞음
    

for i in range(int(input())):
    n = int(input())
    while True:
        if prime(n): #n이 소수이면(함수 실행 결과가 True이면)
            print(n) #n출력
            break
        else: #n이 소수가 아니면
            n += 1 #n에다 += 1해줌

온갖 방법을 다 해봤는데 시간초과로 해결되지 못 했다

그래서 소수를 판별하는 함수를 만들어서 푸니까 해결되었다...