정수 제곱근 판별 (with Python3)

https://programmers.co.kr/learn/courses/30/lessons/12934

코딩테스트 연습 - 정수 제곱근 판별

checkInt 함수는 정수인지 아닌지 확인한 후 정수라면 1을 더한 값을 제곱해서 리턴하고 정수가 아니면 -1을 리턴한다.

solution함수에서 checkInt에 전달 인자로 sqrt(n), 즉 n의 제곱근을 주면서 그 제곱근 값이 정수라면 그 수는 제곱수가 되므로 1을 더한 값을 제곱해서 리턴하게 된다.

from math import sqrt

def solution(n):
    return checkInt(sqrt(n))

def checkInt(x):
    return (x + 1) ** 2 if x == int(x) else -1