728x90
https://programmers.co.kr/learn/courses/30/lessons/12934
프로그래머스에서 레벨1 연습문제를 왕창 풀다가 기억해두면 좋을 것 같은 문제라서 따로 포스팅을 하게 되었다.
정수 제곱급 판별 문제로 math 모듈을 쓰면 엄청 편하다!
정답 코드
import math
def solution(n):
if math.sqrt(n).is_integer() :
return (math.sqrt(n)+1)**2
else :
return -1
math.sqrt(x) 함수는 x에 루트를 씌운 값을 반환한다.
반환된 값이 정수인지 아닌지로 판단해서 정수가 아니면 제곱근이 아니기 때문에 -1을 반환고, 정수라면 제곱근이기 때문에 n+1의 제곱을 반환한다.
import math를 사용하지 않는 방법도 찾아봤는데 복잡하고 코드도 길어져서 확실히 모듈을 사용하는 것이 좋은 것 같다.
728x90