본문 바로가기
Programmers Coding-Test

[Python] 연습문제 : 정수 제곱근 판별

by 으녜 2021. 4. 11.
728x90

 

 

코딩테스트 연습 - 정수 제곱근 판별

임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다. n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함

programmers.co.kr

 

💡문제💡


 

정수 n이 어떤 정수 x의 제곱이면 x+1의 제곱을 리턴하고, 아니라면 -1을 return 하시오.

 

 

[제한 사항]

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

 

[예시]

n = 121일때, 121은 정수 11의 제곱이므로 (11+1)을 제곱한 144를 return 합니다.

 

 

코드 및 풀이👩🏻‍💻


'''

import math

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

'''

 

1 ) 제곱근을 int형이면 즉, 어떤 정수의 제곱이면 x+1의 제곱을 return

 

728x90