math
모듈의 sqrt
함수를 이용하면 제곱근을 구할 수 있다.sqrt
함수를 사용하면 값이 float
로 반환되기 때문에 int
로 묶어서 형변환을 했다.if else
조건문을 사용해 1로 나눈 나머지가 0 이면 제곱근이고 아니면 제곱근이 존재하지 않는거로 간주하도록 만들었다.# programmers, phase1 : 정수 제곱근 판별, python
from math import sqrt
def solution(n):
return int(sqrt(n) + 1) ** 2 if sqrt(n) % 1 == 0 else -1
https://programmers.co.kr/learn/courses/30/lessons/12934
github