[Programmers] 정수 제곱근 판별

Python

import math
def solution(n):
    if math.sqrt(n) == int(math.sqrt(n)):
        return (math.sqrt(n) + 1) ** 2
    return -1