[Python] 프로그래머스 정수 제곱근 판별

https://school.programmers.co.kr/learn/courses/30/lessons/12934?language=python3

프로그래머스

풀이

Sol1 은 math 라이브러리의 sqrt 함수를 사용해서 풀었고 Sol2는 for문을 사용해 풀었다.

 

테스트케이스 통과 시간은 for를 사용하는 sol2가 더 길다.

from math import * 

def solution(n):
    # Solution 1
    if int(sqrt(n)) == sqrt(n):
        return (sqrt(n) + 1)**2
    else:
        return -1
    
    # Solution 2
#     for i in range(1, int(sqrt(n))+1):
#         if i*i == n:
#             return (i+1)**2
    
#     return -1