끄적끄적 코딩일지

[Programmers]정수 제곱근 판별(난이도:★★★) 본문

알고리즘

[Programmers]정수 제곱근 판별(난이도:★★★)

BaekGyuHyeon 2022. 5. 16. 19:30

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

 

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

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

programmers.co.kr

class Solution {
    public long solution(long n) {
        // long answer = (long)Math.sqrt(n);
        long answer = solutionA(n);
        if(answer * answer != n){
            return -1;   
        }else{
            return (answer+1L) * (answer+1L);
        }
        
    }
    // Math.sqrt() 구현해보기
    // 입력한 정수를 소인수분해하여 제곱근인지 판별
    // 만약 64를 입력하면 64 = 2 * 2 * 2 * 2 * 2 * 2로 표현된다.
    // 여기서 2가 2번째로 나올때마다 곱하면 res = 2 * 2 * 2 -> 8
    public long solutionA(long n){
        long tmp = n;
        long contain = 0L;
        long res = 1L;
        while(tmp > 1L){
            for(long i = 2L; i<= tmp; i += 1L){
                if(tmp % i == 0L){
                    if(contain == 0L){
                        contain = i;
                    }else if(contain == i){
                        contain = 0L;
                        res *= i;
                    }
                    tmp /= i;
                    break;
                }
            }
        }
        return res;
    }
}