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

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

python

import math


def solution(n: int) -> int:
    square_root = math.sqrt(n)

    if square_root - int(square_root) == 0:
        return (int(square_root) + 1) ** 2
    else:
        return -1