프로그래머스 / 코딩 테스트 / 정수 제곱근 판별

문제 설명

 

 

문제 해결

 

정수 n의 제곱근(x)을 구하고, x가 n의 제곱근이 맞다면 x + 1의 제곱을 구하여 반환하는 함수를 만들면 된다.

 

math 모듈의 sqrt 함수를 사용하여 n의 제곱근을 구한다.

위키독스

int형태로 변환된 제곱근이 실제로 n의 제곱근이 맞는지 확인한다.

그리고 문제에서 요구한 대로 제곱근 + 1 ** 2의 값을 반환한다.

 

import math

def solve(n):
    x = int(math.sqrt(n))
    if x ** 2 !=n:
        return -1
    else:
        return (x+1)**2
    

def solution(n):
    answer = solve(n)
    return answer