Python_제곱 판별

김보람·2022년 4월 13일
0

Python

목록 보기
4/39

<문제>

임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.
n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.

<풀이>

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

<다른 풀이>

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

<틈새 공부>

찾는 값의 형태가 int인지 float인지 확인하고자 한다면 i%1 == 0:을 이용하자.

profile
starry_developer

0개의 댓글