Algorithm/Programmers

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

뭐든 해보기 2023. 5. 16. 17:21

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

풀이)

1. k는 제수. 2로 선언

2. 중복 값 허용 안하는 HashSet에 담으려고 set 선언

3. n이 1 이상일 때 실행되는 반복문으로 k가 n으로 나눠질 경우 k를 set에 담고 n을 n/k의 몫으로 치환

4. k가 n으로 나눠지지 않을 경우에는 k의 값을 증가시켜줌

5. set을 ArrayList로 변환 후 오름차순 정렬 후 해당 list 리턴

import java.util.Set;
import java.util.HashSet;
import java.util.Collections;
import java.util.ArrayList;
class Solution {
    public ArrayList solution(int n) {
        int k = 2;
        Set<Integer> set = new HashSet<>();
		while(n>1) {
			if(n%k==0) {
				set.add(k);
				n = n/k;
			}
			else {
				k++;
			}
		}
        ArrayList<Integer> arrlist = new ArrayList<>(set);
		Collections.sort(arrlist);
        return arrlist;
    }
}

문제에 오름차순으로 담으라는 설명 제대로 안보고 정렬 안넣었다가 왜 안되는건지 생각했다.

문제를 제대로 읽자..