[프로그래머스/JAVA] 정수 제곱근 판별

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

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

 

 

n이 양의 정수 x의 제곱이라면 (x+1)^2 을 리턴하고, 아니라면 -1을 리턴하는 문제다.

처음에 Math 클래스가 제공하는 메소드를 사용하지 않고 풀었더니 몇개의 테스트 케이스에서 시간초과가 떴다.

 

두 가지 메소드를 알고 있으면 간단하게 풀 수 있다.

 

* 제곱하기 : Math.pow(double a, double n)

* 제곱근 구하기 : Math.sqrt(double a)

 

Math 클래스가 제공하는 메소드는 모두 정적이므로 Import 선언을 하지 않아도 된다.

 

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

 

pow(), sqrt() 메소드는 모두 double형을 반환하기 때문에 문제 형식에 따라서 long형으로 캐스팅하는 과정이 필요하다.

제곱근이 양의 정수가 아닌 경우에는 루트값을 생각하면 된다. (입출력 예 n=3일 경우 n의 제곱근은 3^1/2)