[Coding Test Bookmark]n의 배수 고르기(steam사용)

문제 설명

정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ n ≤ 10,000
1 ≤ numlist의 크기 ≤ 100
1 ≤ numlist의 원소 ≤ 100,000

입출력 예

n numlist                                   result
3 [4, 5, 6, 7, 8, 9, 10, 11, 12] [6, 9, 12]
5 [1, 9, 3, 10, 13, 5]                [10, 5]
12 [2, 100, 120, 600, 12, 12] [120, 600, 12, 12]

입출력 예 설명

입출력 예 #1
numlist에서 3의 배수만을 남긴 [6, 9, 12]를 return합니다.

입출력 예 #2
numlist에서 5의 배수만을 남긴 [10, 5]를 return합니다.

입출력 예 #3
numlist에서 12의 배수만을 남긴 [120, 600, 12, 12]를 return합니다.

 

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

 

        arr.stream().mapToInt(i->i).toArray();

간단히 말하면, ArrayList(List타입)를 int[](배열)로 변환하는 코드.

먼저 ArrayList를 stream으로 변환.

mapToInt(i -> i)를 통해 각 요소를 int로 매핑.

최종적으로 toArray()로 int배열로 변환