[프로그래머스] 정수 제곱근 판별 (Python)

문제 링크 (Level 1)

https://school.programmers.co.kr/learn/courses/30/lessons/12934?language=python3 

프로그래머스

 

나의 코드 (Python)

def solution(n):
    answer = 0
    x = n ** 0.5
    if x == int(x): # 제곱근 한 x가 int()를 씌었을때 같으면 그건 정수
        answer = (x+1) ** 2
    else: # 정수 제곱근이 아닌 경우
        answer = -1
    return answer

 

n을 제곱근 한 x에 int로 형변환 했을 때 값이 같다면 x는 정수입니다.

따라서 정수 제곱근 입니다.