[JAVA] 정수 제곱근 판별

문제

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

프로그래머스

 

정답(100)

import java.util.*;

class Solution {
   public long solution(long n) {
        long answer = -1;
        long tmp  = (long) Math.sqrt(n);
        if(tmp*tmp == n && tmp > 0){
            answer = (tmp+1) * (tmp+1);
        }

        return answer;
    }
}

 

첫번째 시도(61.1)

import java.util.*;

	class Solution {
		  public long solution(long n) {
			long answer = -1;
			for(int i=1; i*i<=n; i++){
				if(i*i == n){
					answer = (i+1) * (i+1);
				}
			}
			return answer;
		}
	}

- 처음에는 속도 for문의 속도문제인가? 싶었다. 하지만 나중에 정답을 푼 후에 for문안의 i를 long으로 바꿔야 한다는 사실을 알고 난 후 시도 해봤더니 100점이 나왔다.

 

첫번째 시도의 정답(100)

import java.util.*;

	class Solution {
		  public long solution(long n) {
			long answer = -1;
			for(long i=1; i*i<=n; i++){
				if(i*i == n){
					answer = (i+1) * (i+1);
				}
			}
			return answer;
		}
	}

 

 

두번째 시도(61.1)

import java.util.*;

class Solution {
      public long solution(long n) {
        long answer = -1;
        int tmp  = (int) Math.sqrt(n);
        if(tmp*tmp == n){
            answer = (tmp+1) * (tmp+1);
        }

        return answer;
    }
}

- 첫번째 시도와 다르게 Math.sqrt를 통해 제곱근을 구한 후 시도 하였지만 첫번째 시도와 같은 점수가 나왔다. 뭐가 문제일지 생각해 보니 문제의 제한사항에 보니

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

라고 쓰여져 있는 것이다. int로는 제곱근이 나올수 없는 큰 양의 정수이기 때문에 제곱근도 데이터 유형을 long으로 잡아야 한다고 느낀후 다시 시도해 본 후 정답이 나왔다. (정답은 맨 위에 있음)