20. 정수 제곱근 판별

유형 : 구현 

풀이 시간 : 10분 이내

 

def solution(n):
    answer = 0
    x = 1
    while True:
        answer = x**2
        if answer == n:
            answer = (x+1)**2
            break
        if answer > n:
            answer = -1
            break
        x += 1 
    return answer

 

종료 조건은 2가지 : answer이 n보다 커질 때, 그리고 answer이 n과 같을 때.

answer이 더 커지는 경우 제곱수가 아니므로 answer = -1로 설정하고 break

answer와 같을 경우, (n+1)**2 반환