[프로그래머스] 소인수분해 (Java)

문제 설명

소인수분해란 어떤 수를 소수들의 곱으로 표현하는 것입니다. 예를 들어 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 합니다.

풀이

//List, ArrayList 사용을 위해 import
import java.util.List;
import java.util.ArrayList;

class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();
        
        //자연수 n의 소인수를 구하기 위한 반복문
        for(int i = 2; i <= n; i++) {
            /*나누어 떨어지면 i는 소인수, 
            몫이 2이상인 경우도 있기 때문에 안 나눠질 때까지 나눠줌*/
            if(n % i == 0) {
                while(n % i == 0) {
                    n /= i;
                }
                list.add(i); //구한 소인수를 list에 넣어줌
            }
        }
        int[] answer = new int[list.size()]; //소인수 갯수와 같은 길이로 설정
        //list의 값들을 answer 배열에 넣어줌
        for(int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }
        return answer; //소인수 배열 return
    }
}