[프로그래머스] 정수 제곱근 판별 (파이썬)

 

https://programmers.co.kr/learn/courses/30/lessons/12934

코딩테스트 연습 - 정수 제곱근 판별

 

 

---

 

 

<나의 풀이>

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

# 2번 풀이
def solution(n):
    sqrt = 1
    while sqrt <= n :
        if (sqrt ** 2) == n :
            return (sqrt+1)**2
        sqrt += 1
    return -1

 

1번 풀이는 math 모듈의 sqrt 함수를 사용해서 풀었고,

2번 풀이는 제곱근의 정의를 이용하여 풀었다.

 

 

 

<다른 사람의 풀이>

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

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

# <배운점> 
# => n^(1/2)은 제곱근..! 모듈을 안써도 된다는 것을 배움