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

문제 설명

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

제한사항

• 2 ≤ n ≤ 10,000

입출력 예

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

입출력 예 설명

입출력 예 #1

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

17은 소수입니다. 따라서 [17]을 return 해야 합니다.
입출력 예 #3

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

Solution

import java.util.*;

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

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

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

        return answer;
    }
}

해설

loop를 돌면서, 나눠진 경우 이미 저장된 값인지 확인한다.

1. 이미 저장된 경우 : 다음 i로 넘어간다.
2. 저장되어 있지 않은 경우 : i를 list에 저장한다. n을 i로 나눈다. i를 다시 원래 상태로 확인하여 그 수로 나눠지는 지 확인한다.