CS/Algorithm

파이썬 실수형 변수가 정수형인지 실수형인지 확인

Hinos 2020. 12. 19. 22:44
반응형

1. 문제 요약

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

import math

def isInteger(strFloat):
    
    try:
        res = int(strFloat)
        return True
    except:
        return False

number = 121

fRes = math.sqrt(number)

answer = 0

strRes = str(fRes)
if isInteger(fRes):
	iRes = int(fRes)
	answer = (iRes + 1) * (iRes + 1)
else :
	answer = -1
	
print(answer)
반응형