소인수분해

문제

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

조건

• 2 ≤ n ≤ 10,000

예시

n	result
12	[2, 3]
17	[17]
420	[2, 3, 5, 7]

풀이

• 소인수의 개수를 미리 알 수 없으므로 자료구조의 크기를 동적으로 바꿀 수 있고 순서를 가지는 List 구조를 사용
• for문의 반복을 사용해 i = 2부터 차례로 n을 나눌 수 있는지 확인
• 만약 n이 i로 나누어진다면 i는 n의 인수이므로 List에 추가하는데, 추가하기 전 n을 나눌 수 있을 때까지 나눈다
  (인수의 갯수를 함께 기록하는 것이 아니라 종류만 기록하기 때문에 i가 인수라면 나머지가 0이 아닐 때까지 n을 나눈다)
• 차례로 i를 1씩 증가하다보면 조건식이 거짓이 되어 반복문은 종료된다
• List의 size와 같은 길이를 갖는 int 배열을 생성한 뒤, 차례로 List의 값을 넣어 반환한다

코드

import java.util.*;

class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();

        for (int i = 2; i <= n; i++) {
            if (n % i == 0) {
                while (n % i == 0) {
                    n /= i;
                }
                list.add(i);
            }
        }

        int[] answer = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }

        return answer;
    }
}