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

사용언어 : JAVA

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

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

 

문제

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

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

나의 코드

public class Sqrt {
    class Solution {
        public long solution(long n) {
            long answer = 0;
            long num = 0;
            //n이하수만큼 나눠서 n으로 나눴을 때 i가 나오고 나머지가 0인 값을 찾기
            for(int i = 1; i<=n; i++){
                if((n/i) == i && n%i == 0){
                    num = i;
                    break;
                }else{ //아닌 경우
                    num = -1;
                }
            }
    
            if(num != -1){ //맞으면 1 더해서 제곱
                num = (num + 1) * (num + 1);
            }
    
            answer = num;
            
            
            return answer;
        }
    }
}

 

입출력 예

n	return
121	144
3	-1

 

코드설명

1. 제곱근인 지 판별하기 위해 n만큼 반복한다.

2. n 나누기 i 가 i 이면서 나머지가 0인 경우 break

3. 아닌 경우 -1 한다. 

4. break로 받아온 i값에 1을 더해 제곱한다. 

5.num에 담겨져 있는 값을 answer에 담아 리턴