입문 - 소인수분해

입문 - 소인수분해

문제설명

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

---

제한사항

• 2 ≤ n ≤ 10,000

---

입출력 예

 

---

입출력 예 설명

입출력 예 #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를 이용

>> 매개변수 n을 1부터 n까지 나누면서 나머지가 0인 값을 찾고, 그 값이 소수인지 확인하기 위해 isPrime 메서드를 이용한다.

 

>> isPrime 메서드는 매개변수 n이 2보다 작으면 false를 반환한다.

>> n이 2보다 크거나 같을 때는 반복문을 통해 n을 i로 나누면서 나머지가 0인 값을 찾고 false를 반환한다.

>> 그 외의 경우는 true로 반환한다.

 

>>  소수라면 numList에 값을 넣어준다.

>> answer배열을 만들고 각 인덱스에 numList에 해당하는 인덱스를 넣어주고 answer를 반환한다.

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

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

        for (int i = 1; i <= n; i++) {
            if (n % i == 0) {
                if (isPrime(i)) {
                    numList.add(i);
                }
            }
        }
        int[] answer = new int[numList.size()];
        for (int i = 0; i < numList.size(); i++) {
            answer[i] = numList.get(i);
        }
        return answer;
    }
    public boolean isPrime(int n) {
        if (n < 2) {
            return false;
        }

        for (int i = 2; i <= Math.sqrt(n); i++) {
            if (n % i == 0) {
                return false;
            }
        }
        return true;
    }
}

 

https://school.programmers.co.kr/learn/courses/30/lessons/120852