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

프로그래머스

띵지니어 2023. 6. 20. 18:58

프로그래머스

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

programmers.co.kr

def solution(n):
    a = n ** 0.5
    if a == int(a):
        return (a+1) ** 2
    else:
        return -1

n ** 0.5 을 하면 float 타입이 반환이 된다.

루트 씌운 n 값을 정수형(n) 값과 비교해서 같으면 정수 제곱근이다.

ex ) n = 121 일때

√n = 11.0

11.0 == int(11.0) True -> 정수 제곱근

ex ) n = 50

√n ≒ 7.071...

7.071 == int(7.071) False -> 정수 제곱근 아님