[프로그래머스] Lv.1 정수 제곱근 판별_파이썬

 

정수 제곱근 판별입니다!

입력받은 n이 x의 제곱이라면, x+1의 제곱 값을 구하고,

n이 양의 정수 x의 제곱이 아니면, -1을 리턴하는 문제입니다!

 

정수 제곱근 판별

• 입력받은 n에 제곱근을 씌워줍니다.
  -> 이때 n의 제곱근의 값이 정수가 아니면 너무 큰 소수가 나올 수 도 있으므로 소수 둘째 자리에서 반올림을 했습니다

• 이 n이 정수이면, n의 제곱근 값에 1을 더하고 제곱을 하였고,
• 이 n이 소수이면, -1을 출력합니다.

def solution(n):
    n = round(n**(0.5),1)    
    if n == int(n):            
        answer = (n+1)**2
    else :
        answer = -1
    return answer

 

그리고 파이써닉하게 한번 바꾸어 보았습니다! 

 

def solution(n):
    n = round(n**(0.5),1)    
    answer = (n+1)**2 if n == int(n) else  -1             
    return answer

 

핫.. 뿌듯..한 기분이 듭니닷.. 👨‍💻