[python3][level 1] 정수제곱근판별하기

루트를 이용해 루트가 정수값일 경우 x + 1 의 제곱을 리턴하고 아닐경우 no를 리턴하자

import math

def nextSqure(n):
    if math.sqrt(n) == int(math.sqrt(n)):
        return int((math.sqrt(n) + 1) ** 2)
    return 'no'
# 아래는 테스트로 출력해 보기 위한 코드입니다.
print("결과 : {}".format(nextSqure(121)));