[Algorithm] [Python] Programmers - 정수 제곱근 판별

정수 제곱근 판별

---

문제 설명

---

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

제한 사항

---

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

입출력 예

---

n	return
121	144
3	-1

121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴한다.

3은 양의 정수의 제곱이 아니므로, -1을 리턴한다.

제출 코드

---

import math

def solution(n):
   x = math.sqrt(n)
   if x % 1 == 0:
       answer = (x+1)**2
   else:
       answer = -1
   return answer

코드 설명

---

제곱근을 수월하게 구하기 위해 math module을 import한다.

그 이후, math.sqrt()를 이용하여 제곱근을 구하여 x에 저장한다.

만일 x가 양의 정수인 경우 %1==0 조건을 만족할 것이며 그 외 만족하지 못할 것 이다.

따라서, 해당 조건을 통해 분기를 진행하여 적절히 answer값을 초기화 한 이후 return하도록 한다.

다른 사람의 풀이

---

def solution(n):
 sqrt = n ** (1/2)
 if sqrt % 1 == 0:
   return (sqrt + 1) ** 2
 return -1

위 코드의 경우, 구한 제곱근이 양의 정수인지 판별하는 로직은 필자가 짠 코드와 같다.

그러나 math 모듈을 import하지 않고 sqrt = n ** (1/2) 를 통하여 제곱근을 구해냈다..

제곱의 역연산 개념을 이용해 1/2를 곱해줬다.

정말 간단한 내용임에도 불구하고 문제를 풀 당시 전혀 생각하지 못했었다..

또한, 기존 문제 풀이 사이트에서 제공되는 code foramt을 건들여도 되는건지 몰랐다.

문제 풀이 페이지에 접근하면 answer라는 변수가 초기화 되어있는데 굳이 안써도 되는것 같다.