Programmers/Level 1

C++ / 프로그래머스 / 정수 제곱근 판별

GitHubSeob 2021. 9. 13. 19:10

문제

https://programmers.co.kr/learn/courses/30/lessons/12934

 

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

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

programmers.co.kr

문제풀이

제곱근을 구해주는 sqrt함수를 이용해서 n의 제곱근을 구한다.

n의 제곱근이 정수형인지를 체크해주고 정수형이라면 제곱근 값+1의 제곱을 한다.

소수점이 있다면 -1을 return 한다.

 

코드

#include <string>
#include <vector>
#include <cmath>
using namespace std;

long long solution(long long n) {
     long long answer = 0;
    if(sqrt(n)==(int)sqrt(n)) answer = pow(sqrt(n)+1, 2);
    else answer = -1;
    return answer;
}