https://school.programmers.co.kr/learn/courses/30/lessons/120909
import math
def solution(n):
return 1 if math.sqrt(n) == int(math.sqrt(n)) else 2
n
이 된다는 뜻이니까, 제곱근과 정수 변환된 값이 같다면 1
을 반환하고, 아니라면 제곱수가 아니므로 2
를 반환했다.def solution(n):
return 1 if (n ** 0.5).is_integer() else 2
is_integer()
함수를 활용하여 해당 값이 정수인지를 확인했다.math.sqrt()
를 통해 제곱근을 구할 수 있지만, 이는 n의 1/2 제곱
이므로 n ** 0.5
로 표현할 수 있다.math.sqrt()
함수는 항상 양의 제곱근만 반환하므로, 음의 제곱근이 필요하다면 앞에 부호를 붙여서 반환받자.피드백은 언제나 환영입니다 :)