1. 정수 제곱근 판별
URL : https://programmers.co.kr/learn/courses/30/lessons/12934
1.1문제 설명
- 임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.
- n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.
1.2 제한 사항
- n은 1이상, 50000000000000 이하인 양의 정수입니다.
1.3 Solution
1
2
3
4
5
6
7
8
import math
def solution(n):
a = int(math.sqrt(n))
if a * a == n:
return (a + 1) * (a + 1)
else:
return -1
1
2
n= 515
solution(n)
1
-1
1
2
n= 144
solution(n)
1
169
- math 패키지의 sqrt 메소드를 사용하여 제곱근을 찾아 정수로 변경하여 다시 제곱하면 n으로 나오는지 확인
- 만일 True으로 나온다면 각 제곱근에 + 1을 한 뒤 제곱을 함
- 만일 False로 나온다면 -1을 리턴함