[Java 알고리즘 문제] (68) 소인수분해

Feb 08, 2025

school.programmers.co.kr

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

문제 설명

소인수분해란 어떤 수를 소수들의 곱으로 표현하는 것입니다. 예를 들어 12를 소인수 분해하면 2 * 2 * 3 으로 나타낼 수 있습니다. 따라서 12의 소인수는 2와 3입니다. 자연수 n이 매개변수로 주어질 때 n의 소인수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

2 ≤ n ≤ 10,000

해답


import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

class Solution {
    public int[] solution(int n) {
        Set<Integer> factors = new HashSet<>();
        while (n % 2 == 0) {
            factors.add(2);
            n /= 2;
        }
        for (int i = 3; i <= Math.sqrt(n); i += 2) {
            while (n % i == 0) {
                factors.add(i);
                n /= i;
            }
        }
        if (n > 2) {
            factors.add(n);
        }
        int[] result = new int[factors.size()];
        int index = 0;
        for (int factor : factors) {
            result[index++] = factor;
        }
        Arrays.sort(result);
        return result;
    }
}