[Programmers] 소인수분해(자바)

➰문제

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

 

 

 

---

 

💡풀이

⭐제출 코드

import java.util.List;
import java.util.ArrayList;

class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();
        List<Integer> prime = new ArrayList<>();
        int count = 0;
        
        // n의 약수 찾기
        for(int i=2; i<=n; i++){
            if(n%i == 0){
                list.add(i);
            }            
        }
        
        // n의 약수 중에서 소수 찾기
        for(int i=0; i<list.size(); i++){
            for(int j=2; j<=n; j++){
                if(list.get(i)%j == 0){
                    count += 1;
                }
            } 
            // 약수가 자신 하나라면
            if(count == 1){
                prime.add(list.get(i));
            }
            count = 0;
        }
        
        int[] answer = new int[prime.size()];
        // list의 값들을 answer배열에 넣어줌
        for(int i=0; i<answer.length; i++){
            answer[i] = prime.get(i);
        }
        return answer;
    }
}

ArrayList를 활용하여 작성했다.

 

🧐 코드 해석

1. n의 약수를 찾아서 list에 넣어준다.

2. n의 약수 중에서 2 ~ n까지 나누었을 때 나누어 떨어진다면 count를 1증가 시킨다.

3. count가 1 이라는 것은 나누어 떨어지는 수가 자기 자신 하나라는 뜻! 이므로 소인수를 담을 prime 리스트에 넣어준다.

4. 소인수 값을 return 할 answer 배열을 prime의 길이만큼 생성한다.

5. 리스트 prime의 값을 answer에 하나씩 넣어주고, answer을 return 한다.