Algorithm/Programmers(Java)

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

비망노트 2022. 12. 7. 22:34
문제 설명

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

 

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

 

입출력 예
n return
121 144
3 -1

 

입출력 예 설명

입출력 예 #1

121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.

 

입출력 예 #2

3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

 

⭕ 풀이

class Solution {
    public long solution(long n) {
        long answer = -1;

        double sqrt = Math.sqrt(n);
        int floor = (int)Math.floor(sqrt);

        if(sqrt==floor){
            answer = (long) Math.pow(floor+1,2);
        }

        return answer;
    }
}

 

✅ 흐름

1. Math.sqrt함수를 사용해 입력값 long타입 n의 제곱근을 구해 double타입 변수 sqrt에 담는다

2. sqrt의 소수점이하를 자르고 정수형으로 바꿔 floor변수에 담는다.

3. 이 둘을 비교한다.

입출력 예로 테스트해보면

// 입출력 예1
121
sqrt : 11.0
floor : 11

// 입출력 예2
3
sqrt : 1.7320508075688772
floor : 1

11.0과 11이 같냐고 물으면 true를 반환해주므로 sqrt == floor 를 비교한다

 

4. 동일하다면 floor에 1을 더한값의 제곱을 answer에 담는다.

4-1. 동일하지 않았다면 answer의 값은 -1 그대로, 제곱근이 없다는것이다.

 

5. answer를 출력한다.

 

 

 

 

-출처

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