programmers #정수 제곱근 판별

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

 

입출력 예

nreturn

121	144
3	-1

 

통과 코드)

def solution(n): 
    for x in range(1, n+1):
        if n == x**2:
            return (x+1)**2
        
    return -1

 

 

def solution(n): 
    for x in range(1, n):
        if n == x**2:
            return (x+1)**2
        else:
        	return -1

맨 처음에 생각해냈던 코드이다. 여기서 에러가 난 이유는 else: return -1 때문!

만약 n = 121이더라도 if 밑에 else가 있기 때문에 121 == 1**2 부터 바로 if를 성립하지 않아 else로 넘어가서 -1을 출력했었다.

그래서 return -1은 아예 if 밖으로 ㄱㄱ

 

그리고 for x in range(1, n):

이 틀린 이유는 만약 n = 1 이라면 x와 비교가 되지 않기때문에 에러가 낫던 듯!

오늘도 열심히다! 기특하군