[programmers] 소인수분해 - Java

문제 설명

제한사항

• 2 ≤ n ≤ 10,000

입출력 예

입출력 예 설명

• 12를 소인수분해하면 2 * 2 * 3 입니다. 따라서 [2, 3]을 return합니다.

• 17은 소수입니다. 따라서 [17]을 return 해야 합니다.

• 420을 소인수분해하면 2 * 2 * 3 * 5 * 7 입니다. 따라서 [2, 3, 5, 7]을 return합니다.

solution.java

import java.util.*;

class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();
        
        // 2의 배수인지 확인
        while (n % 2 == 0) {
            if (!list.contains(2)) {
                list.add(2);
            }
            n /= 2;
        }
        
        // 3부터 n의 제곱근까지의 홀수 중 약수가 있는지 확인
        for (int i = 3; i <= Math.sqrt(n); i += 2) {
            while (n % i == 0) {
                if (!list.contains(i)) {
                    list.add(i);
                }
                n /= i;
            }
        }
        
        // 홀수와 짝수 중 약수가 없다면 소수
        if (n > 2) {
            list.add(n);
        }
        
        return list.stream().mapToInt(e -> e).toArray();
    }
}

핵심 키워드

• 처음 시도했던 방식은 n을 2부터 n까지 1식 증가시키며 나머지를 계산하는 코드를 작성했다.
• 해당 방식은 값을 구할 수는 있었지만 실행 시간이 10초를 넘어 효율적인 코드라고 생각하지 않았다.

• 수정한 코드는 먼저 2의 배수인지 확인을 한 다음, 3부터 2씩 증가시킨 값을 나머지 연산해서 약수가 있는지 판단하였다.
• 또한 반복문의 범위가 n까지가 아닌 n의 제곱근까지로 설정하였는데, n을 제곱근보다 큰 수로 나눌 수 있는 경우, 다른 인수는 이미 찾은 제곱근보다 작을 것이기 때문이다.

결론!