[프로그래머스] Level1 - 정수 제곱근 판별 (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을 리턴합니다.

내 문제 풀이

• 2가지로 생각. 하나는 math에서 제곱근 함수인 sqrt를 이용해서 푸는 방법
• 하나는 sqrt 안 쓰고 반복문만 써서 푸는 방법

(1) sqrt 쓰는 풀이

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

• 주어진 숫자 n이 제곱근으로 나눴을 때 나머지가 0이면 n을 제곱근으로 나눈 값 (당연히 제곱근이겠지...ㅋ) 에 1을 더해서 제곱한 값을 return. 만약 주서진 숫자 n이 제곱근으로 나누어 떨어지지 않으면 -1 return

(2) sqrt 안 쓰는 풀이

def solution(n):

    cnt = 1 #1부터 시작해서 제곱한 값이 n과 같은지 비교하는 변수
    
    while(cnt**2 <= n):
        if cnt** 2 == n:
            return (cnt+1)**2
        else:
            cnt += 1
    return -1

• cnt의 제곱한 값인 cnt ** 2 가 주어진 숫자 n 보다 작거나 같은 범위 내에서 반복문 실행
• 만약 cnt ** 2의 값이 n과 같다면 그 즉시 cnt+1의 제곱한 값인 (cnt+1)**2를 return
• 만약 cnt ** 2의 값이 n과 같지 않다면 cnt의 값을 1 증가시킴.
• 만약 반복문을 빠져나온 경우, 즉 cnt ** 2의 값이 n보다 클때는 더이상 비교 자체를 할 필요가 없기 때문에 -1을 return

아쉬운 점

• 많은 사람들이 공감한 방법을 보니까 내 풀이의 (1)번 방법을 sqrt를 쓰지 않고 풀었다. 즉 sqrt = n ** (1/2) 로 설정하고 푼 것. ** 연산자의 값을 무조건 정수만 들어가야 한다고 생각했었는데 분수도 되는 걸 알았다