[프로그래머스 Lv0.] 소인수분해 (JAVA)

gayoung·2023년 2월 11일
0

알고리즘

목록 보기
32/50

1. 문제

문제 설명

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

제한사항

  • 2 ≤ n ≤ 10,000

입출력 예

nresult
12[2, 3]
17[17]
420[2, 3, 5, 7]

2. 풀이 과정

내가 생각한 진행 과정

  • n을 2부터 n까지 계속 나누면서 array에 안들어있으면 넣어준다.
    • ex) 12 -> n=12이고 2로 나뉘어지니까 array=[2] -> n=6이고 2로 나뉘어지지만, 이미 array에 있음 -> n=3이고 3으로 나뉘어지니까 array=[2, 3]
  • 마지막에 array가 List(ArrayList)이므로 stream을 이용해 int[]로 바꿔줘야함

최종 코드

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

public int[] solution(int n) {
        int[] answer = {};
        List<Integer> array = new ArrayList<>();

        int idx = 2;
        while (n > 1) {
            if (n % idx == 0) {
                if (!array.contains(idx)) {
                    array.add(idx);
                }
                n /= idx;
            } else {
                idx++;
            }
        }
        
        answer = array.stream().distinct().mapToInt(Integer::intValue).toArray();
        return answer;
    }

3. 알게된 것

  • stream 사용방법

0개의 댓글

관련 채용 정보