[프로그래머스] 정수 제곱근 판별 - python

def solution(n):
    answer = 0
    x = int((n+1)**0.5)
    for i in range(1,x+1):
        if i*i == n:
            answer = (i+1)*(i+1)
    if answer == 0 :
        answer = -1
    return answer

 

처음에 범위를 무식하게 n까지 다돌렸다가 계속 시간초과가 났다.

제곱을 찾는 것 이기 때문에 루트n만큼만 돌리면 되는 것 !