[Level1] 정수 제곱근 판별

문제

양의 정수 n이 어떤 정수 x의 제곱근인지 아닌지 판별하라.

x의 제곱근이라면 (x+1)의 제곱을 출력 아니라면 -1을 출력한다.

 

입력

121

3

 

출력

144

-1

---

1. 변수 num에 n의 제곱근을 int형으로 변환하여 저장한다.

2. if문을 통해 다시 num의 제곱이 n이 맞는지 판단한다

3. 같지않다면 answer에 -1을 같다면 (num+1)에 제곱을 answer에 저장한다.

 

python

def solution(n):
    answer = 0
    num=int(n**0.5)
    if n!=num**2:
        answer=-1
    else:
        answer=(num+1)**2
    return answer