[Python] 정수 제곱근 판별

문제 설명

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

제한 조건

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

입출력 예

n	return
121	144
3	-1

• 입출력 예#1
  121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.

• 입출력 예#2
  3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

Solution

import math

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

처음에 math.sqrt(n)으로만 하고 나서도 되는줄 알았으나 101일때 예외가 발생했다. 이럴수가....생각도 못했다. 소수점으로 나오면 제곱했을 때, 본래의 수가 안되는 줄 알았는데! 역시 여러 케이스가 있음을 기억하고 문제를 해결해야할듯 싶다.

그래서 문제는 다 int형으로 변환시켜서 비교했다. 조건에서 양수만 들어온다고 했으므로 음수 처리는 따로 해주지 않았다.

Other Solution

내가 보려고 올리는 다른 풀이

def nextSqure(n):
    sqrt = n ** (1/2)

    if sqrt % 1 == 0:
        return (sqrt + 1) ** 2
    return -1

• 제곱근은 본래 수의 0.5 제곱한 수와 같다
• 소수를 1로 나눈다면 나머지가 존재한다