TIL/Algorithm
[프로그래머스] 정수 제곱근 판별(easy)
재융
2021. 1. 4. 18:51
반응형
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
반응형