정수 제곱근 판별
문제 설명
임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.
n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.
제한 사항
- n은 1이상, 50000000000000 이하인 양의 정수입니다.
입출력 예
n | return |
121 | 144 |
3 | -1 |
Python 코드
from math import sqrt
def solution(n):
return int(sqrt(n) + 1) ** 2 if sqrt(n) % 1 == 0 else -1
def solution(n):
answer = 0
num = n ** 0.5
if num == int(num):
answer = (num+1) ** 2
else:
answer = -1
return answer
* 참고 링크 : https://jokerldg.github.io/algorithm/2021/04/14/int-sqrt.html
C++ 코드
#include <string>
#include <vector>
#include <cmath>
using namespace std;
long long solution(long long n) {
if (sqrt(n) - (int)sqrt(n) == 0) // n이 정수의 제곱인 경우
return (sqrt(n) + 1) * (sqrt(n) + 1);
return -1;
}
* 참고 링크 : https://greenapple16.tistory.com/129
출처
'코딩테스트 > Programmers' 카테고리의 다른 글
[코딩테스트/Programmers] 22_기능개발 (Lv2.) (0) | 2022.01.17 |
---|---|
[코딩테스트/Programmers] 21_다리를 지나는 트럭 (Lv2.) (0) | 2022.01.17 |
[코딩테스트/Programmers] 19_최대공약수와 최소공배수 (Lv1.) (0) | 2022.01.11 |
[코딩테스트/Programmers] 18_자릿수 더하기 (Lv1.) (0) | 2022.01.11 |
[코딩테스트/Programmers] 17_행렬의 덧셈 (Lv1.) (0) | 2022.01.11 |