프로그래머스/Lv. 1

[프로그래머스 코딩테스트] 정수 제곱근 판별(Java)

Sigfriede 2023. 4. 4. 01:10

  문제 설명

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

 

  제한사항

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

 

  입출력 예

n return
121 144
3 -1
class Solution {
    public long solution(long n) {
        long answer = 0;
        double sqrt = Math.sqrt(n);
        
        if (sqrt % 1 == 0) {
            answer = (long)Math.pow(sqrt + 1, 2);
        } else {
            answer = -1;
        }
        return answer;
    }
}

  Math.sqrt는 n의 제곱근을 구하는 클래스와 메소드입니다. 이 메소드의 출력값은 항상 double형이므로 double형 변수를 만들어주었습니다. x가 정수 제곱근이 맞다면 1로 나눈 나머지가 0으로 맞아떨어집니다. 문제에서는 x+1의 제곱의 반환을 원하므로, 마찬가지로 Math 클래스의 제곱을 계산하는 pow 메소드로 계산해주었습니다.

  Math.pow의 경우 answer가 long형이므로 맞추어 long형으로 캐스팅해주었습니다.

 

Math.sqrt(제곱수)
Math.pow(밑수, 지수)

 

  만일 정수 제곱근이 아니라면 -1을 반환하라고 했으므로 조건에 맞춰 식을 세웠습니다.