[프로그래머스, LV.0] 소인수분해

목차

문제 설명

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

---

 

제한사항

• 2 ≤ n ≤ 10,000

 

import java.util.ArrayList;
import java.util.List;
class Solution {
    public int[] solution(int n) {
        List<Integer> factors = new ArrayList<>();
        
        // 소인수 분해 시작
        for (int i = 2; i <= Math.sqrt(n); i++) {
            while (n % i == 0) {
                if (!factors.contains(i)) {
                    factors.add(i);
                }
                n /= i;
            }
        }

        // 남아있는 n이 소수라면 추가
        if (n > 1) {
            factors.add(n);
        }

        // 결과를 배열로 변환
        return factors.stream().mapToInt(Integer::intValue).toArray();
    }
}

코드 설명

1. 소인수 분해 반복문:
   • i를 2부터 시작하여 입력값 n의 제곱근까지 반복한다.
   • n % i == 0일 경우, i가 n의 소인수라는 의미다.
   • 소인수가 중복되지 않도록 factors 리스트에 존재 여부를 확인하며 추가한다.
2. 소수 판별:
   • 루프 종료 후, 남아 있는 n이 1보다 크면 소수로 판단하여 리스트에 추가한다.
3. 리스트를 배열로 변환:
   • factors.stream().mapToInt(Integer::intValue).toArray()를 사용해 List<Integer>를 int[]로 변환한다.
4. 메인 함수:
   • 테스트 값을 n = 12로 설정하고 결과를 출력한다.

특징

• 중복 소인수를 제거하고 오름차순으로 정렬된 결과를 반환한다.
• 제한사항인 2 ≤ n ≤ 10,000 범위에서 효율적으로 동작하도록 제곱근 범위를 사용했다.