본문 바로가기

Java/Java 알고리즘 프로그래머스

[프로그래머스-LEVEL 1] 11. 정수 제곱근 판별

반응형
  • 정수 제곱근 판별
문제 설명

임의의 양의 정수 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을 리턴합니다.


#문제 풀이 개요

-> 입력받은 수의 제곱근을 구해 +1한 제곱을 리턴한다.

-> 항상 입력받은 수 > 제곱근이므로, 제곱근을 입력받은 수와 같게 설정 후, 증감 연산자를 이용해 1까지 만든다.

 

# double형으로 바꿔서 풀이한 결과 -> 일부 테스트 케이스에서 실패

class Solution {
    public double solution(double n) {
        double answer = 0;
        double s=n;
        while(s!=1){
            if(s==Math.sqrt(n)){
                return answer=Math.pow(s+1,2);
            }
            s--;           
        }
        answer=-1;
        return answer;
    }
}

-> 제한 사항을 통과하지 못한 부분이라고 판단

 

#long형으로 변환해 풀이

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

-> 여전히 실패인 테스트 케이스 존재

 

#n이 1이상이므로, 1인 경우도 판단

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

-> 시간 초과로 인한 테스트 케이스 존재

-> 정확성을 높이기 위한 방안 찾기

 

#제곱근은 항상 양의 정수이므로, int형으로 캐스팅해서 사용

class Solution {
    public long solution(long n) {
        long answer = 0;
        int s=(int)n;
        if(n==1){
            return answer=4;
        }
        while(s!=1){
            if(s==Math.sqrt(n)){
                return answer=(long)Math.pow(s+1,2);
            }
            s--;           
        }
        
        return answer=-1;
    }
}

 

 

+)세련된 풀이

class Solution {
  public long solution(long n) {
      if (Math.pow((int)Math.sqrt(n), 2) == n) {
      //만약, n의 값의 제곱근의 제곱이 n이면
            return (long) Math.pow(Math.sqrt(n) + 1, 2);
            //n의 제곱근+1의 제곱을 리턴
        }

        return -1;
        //그 외는 모두 -1리턴
  }
}
반응형