Algorithm/Programmers

[Python | Level1] 정수 제곱근 판별

dotz0ver 2024. 4. 3. 15:15

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

 

프로그래머스

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

programmers.co.kr


문제 설명

임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.

n의 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.


제한 사항

  • n은 1 이상, 50,000,000,000,000 이하인 양의 정수입니다.

문제 풀이

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