0401 정수 제곱근 판별

문제 설명

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

제한 사항

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

입출력 예

n return
121 144
3 -1

코드

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

풀이

ixi가 n보다 작거나 같을 때까지 for문을 계속 돌려준다.
a에 ixi를 넣어주고 a가 n과 같을 경우 a는 양의 정수이므로 answer에 i+1의 제곱값을 넣어 리턴하고 만약 a가 0이 아닐 경우 answer에 -1을 넣어 리턴해준다.