Programmers Level 0 - 약수 구하기

문제 설명

정수 n이 매개변수로 주어질 때, n의 약수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ n ≤ 10,000

 

입출력 예

n	result
24	[1, 2, 3, 4, 6, 8, 12, 24]
29	[1, 29]

입출력 예 설명

입출력 예 #1

• 24의 약수를 오름차순으로 담은 배열 [1, 2, 3, 4, 6, 8, 12, 24]를 return합니다.

입출력 예 #2

• 29의 약수를 오름차순으로 담은 배열 [1, 29]를 return합니다.

 

class Solution {
    public int[] solution(int n) {
        int count = 0;
        for(int i=1;i<=n;i++) {
            if(n%i==0) count++;
        }
        int[] answer = new int[count];
        count = 0;
        for(int i=1;i<=n;i++) {
            if(n%i==0) answer[count++] = i;
        }
        return answer;
    }
}

import java.util.List;
import java.util.ArrayList;
class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();
        for(int i=1;i<=n;i++) {
            if(n%i==0) list.add(i);
        }
        return list.stream().mapToInt(i->i).toArray();
    }
}

리스트를 이용하여 약수를 구하고 stream을 이용하여  int 배열로 변환 하였다.

 

import java.util.stream.IntStream;
class Solution {
    public int[] solution(int n) {
        return IntStream.range(1, n+1).filter(i -> n % i == 0).toArray();
    }
}

IntStream으로 1~n까지의 스트림을 만들고  filter() 로 배수만 취한 후   toArray()로 배열로 만들었다.