[파이썬3 코딩테스트] 정수 제곱근 판별

문제 설명 및 제한 조건

나의 풀이

1 for문을 이용하여 풀이

def solution(n):
    for i in range(1,n+1):
        if i**2 == n:
            return (i+1)**2
        elif i**2 < n:
            i += 1
            continue
        else:
            return -1

2 while문을 이용하여 풀이

def solution(n):
    i=1
    while i>0:
        if i**2 == n:
            return (i+1)**2
        elif i**2 < n:
            i += 1
            continue
        else:
            return -1