[프로그래머스] 정수 제곱근 판별

코딩테스트 연습 - 정수 제곱근 판별

내풀이 (성공)

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

다른방법 1

import math
def nextSqure(n):
    # 함수를 완성하세요
    return 'no' if not math.sqrt(n).is_integer() else (math.sqrt(n)+1)**2

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print("결과 : {}".format(nextSqure(121)));

다른방법 2

import math
def nextSqure(n):
    sqrt = pow(n, 0.5)
    return pow(sqrt + 1, 2) if sqrt == int(sqrt) else 'no'

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print("결과 : {}".format(nextSqure(121)));