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

🔍 알고리즘/프로그래머스 Python

탄치 2022. 9. 29. 00:16

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

import math


def solution(n):
    x = math.sqrt(n)
    xd = x // 1

    if x == xd:
        return (x + 1) * (x + 1)
    else:
        return -1

728x90