[프로그래머스] Lv.0 소인수분해 - 자바(Java)

문제: 코딩테스트 연습 - 소인수분해 | 프로그래머스 스쿨 (programmers.co.kr)

문제설명

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

---

나의 문제 풀이

import java.util.ArrayList;
class Solution {
    public int[] solution(int n) {
        int[] answer = {};
        ArrayList<Integer> list = new ArrayList<>();
        
        for (int i = 2; i <= n; i++) {
			if(n % i == 0) {
				while (n % i == 0) {
					n /= i;
				}
				list.add(i);
			}
		}
        answer = new int[list.size()];
        for (int i = 0; i < answer.length; i++) {
			answer[i] = list.get(i);
		}
        return answer;
    }
}

---

다른 사람의 문제풀이

import java.util.*;

class Solution {
    public int[] solution(int n) {
        List<Integer> primes = new ArrayList<>();
        for (int i = 2; i <= n; ++i) {
            if (isPrime(i) && n % i == 0) {
                primes.add(i);
            }
        }
        int[] answer = new int[primes.size()];
        for (int i = 0; i < answer.length; ++i) {
            answer[i] = primes.get(i);
        }
        return answer;
    }

    private boolean isPrime(int n) {
        for (int i = 2; i <= n / 2; ++i) {
            if (n % i == 0) {
                return false;
            }
        }
        return true;
    }
}

---

문제 출처: 코딩테스트 연습 | 프로그래머스 스쿨 (programmers.co.kr)