[프로그래머스] 정수 제곱근 판별 - 파이썬(Python) - 우당탕탕 개발자 되기 프로젝트

 

프로그래머스

 

1. 문제 설명

2. 풀이과정

1. 매개변수로 받은 n의 제곱근을 우선 구한다. sqrt = n ** (1/2)
2. 만약 구한 제곱근이 양의 정수 즉, 자연수이면 제곱근에 1을 더한 값의 제곱을 리턴한다. 따라서 구한 제곱근을 1로 나눈 나머지가 0이라면 자연수를 뜻한다. if (sqrt % 1 == 0): answer = (sqrt + 1) ** 2
3. 만약 구한 제곱근을 1로 나눈 나머지가 0이 아니라면 구한 제곱근이 실수이므로 -1을 리턴한다. else: answer = -1

반응형

3. 소스코드

def solution(n):
    answer = 0
    
    sqrt = n ** (1/2)
    if (sqrt % 1 == 0):
        answer = (sqrt + 1) ** 2
    else:
        answer = -1
            
    return answer