프로그래머스 Lv1. 정수 제곱근 판별

https://school.programmers.co.kr/learn/courses/30/lessons/12934

프로그래머스

 

문제 임의의 양의 정수 n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴

제한조건 n > 0인 정수 

 

내 답안

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

만약 x와 int(x)가 같다면 x는 정수이므로 이것을 활용했다.