import math
def solution(n):
a = math.sqrt(n)
if a == int(a):
return int((a+1)**2)
else:
return -1
제곱을 어떻게 해야하나 생각하다 라이브러리가 있는지 찾아보니 math.sqrt(n)
를 팢아 사용해보았다.
math.sqrt(n)
는 float타입이라 int로 형을 변환시켜줘야했다.
def solution(n):
a = n ** (1/2)
if int(a) == a:
return int((a + 1) ** 2)
return -1
다른 분이 작성한 답변 중 제곱을 **1/2
로 표현하신 분이 있어 그걸 이용해서 다시 한 번 풀어보았다.