[프로그래머스] 정수 제급곤 판별(feat.정수인지 아닌지 확인하는 법)

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

프로그래머스

 

다음과 같은 흐름으로 문제를 풀었다.

 

1. math라이브러리를 활용하여 판별식 D를 정의 (제곱근)

2. D가 정수라면 D + 1의 제곱을 반환.

3. D가 정수가 아니라면 -1을 반환.

 

import math

def solution(n):
    D = math.sqrt(n)

    if D == int(D):
        return int((D + 1) ** 2)
    else:
        return -1

 

 

math 라이브러리를 사용하지 않고 푸는 방법도 참고하자.

 

def nextSqure(n):
    sqrt = n ** (1/2)

    if sqrt % 1 == 0:
        return (sqrt + 1) ** 2
    return 'no'