S E P H ' S

[Python] 정수 제곱근 판별 본문

Algorithm/Programmers

[Python] 정수 제곱근 판별

yoseph0310 2021. 6. 22. 21:28

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

 

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

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

programmers.co.kr

1. 제곱근과 제곱수를 구하기 위해 import math

2. n이 양의 정수 제곱근이면 n+1의 제곱을, 아니면 -1을 리턴

 

import math
def sol(n):
	return pow(math.sqrt(n)+1, 2) if int(math.sqrt(n)) == math.sqrt(n) else -1