본문 바로가기

TIL/Algorithm

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

반응형

from math import sqrt
def solution(n):
    answer = 0
    if sqrt(n)%1 == 0:
        return int((sqrt(n)+1)**2)
    else:
        return -1
    return answer
    
    
### 다른 문제 풀이
def solution(n):
    sqrt = n ** (1/2)

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