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

안녕하세요 뚜디 입니다:)

코딩테스트 연습 - 정수 제곱근 판별 | 프로그래머스 (programmers.co.kr)

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

---

1. 연습 문제

2. 문제 풀이

3. 소스 코드

4. 결과

---

1. 연습 문제

• 문제 설명

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

• 제한 조건

n은 1이상, 50000000000000 이하인 양의 정수입니다.

• 입출력 예

n	return
121	144
3	-1

2. 문제 풀이

입출력 예#1
121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.
입출력 예#2
3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

문제 풀이는 어렵지 않지만, 아무래도 C/C++ 만하다가 파이썬을 접하다보니 파이썬 답게 풀지를 못하는것 같다.
실행시간이 너무 오래걸린다. 불만족..

파이썬 답게 간결하게 푸신분있으시면 댓글남겨주세요~

3. 소스 코드

import math

def solution(n):
    answer = 0
    for i in range(int(math.sqrt(50000000000000))):
        if(n == i ** 2):
            answer = (i + 1) ** 2
            break;
        else:
            answer = -1

    return answer

4. 결과