[프로그래머스] 연습문제 Lv1. 정수 제곱근 판별 (Python/파이썬)

[프로그래머스 코딩테스트 연습] 연습문제 - Lv1. 정수 제곱근 판별

프로그래머스

문제 설명

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

제한사항

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

입출력 예

n	result
121	144
3	-1

입출력 예 설명

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

 

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

제출

- math 라이브러리의 pow()함수와 sqrt()함수를 사용했다

 math.pow(n, 2) : n의 2 제곱한 결과 반환

 math.sqrt(n) : n의 제곱근 결과 반환

- n의 제곱근인 x가 양의 정수이면 정답 구해주고 아니면 -1 리턴

import math

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

    return answer

 

다른사람의 풀이

- math 라이브러리를 사용하지 않고 ** (1 / 2)를 하여 제곱근을 구해주었다

- if sqrt % 1 == 0 --> 양의 정수일 경우

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

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