[Lv. 0] n의 배수 고르기

📕 문제 설명

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

 

❗ 제한 사항

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

 

✏️ 작성한 코드

import java.util.*;

class Solution {
    public int[] solution(int n, int[] numlist) {
        List<Integer> list = new ArrayList<>();
        
        for(int i = 0; i < numlist.length; i++){
            if(numlist[i] % n == 0) list.add(numlist[i]);
        }
        
        int[] answer = list.stream()
                .mapToInt(Integer::intValue)
                .toArray();
        
        return answer;
    }
}

 

🤔 느낀 점

풀고 났더니 리스트에 추가한 부분도 스트림으로 만들면 편했겠다는 생각을 했다

import java.util.Arrays;

class Solution {
    public int[] solution(int n, int[] numList) {
        return Arrays.stream(numList).filter(value -> value % n == 0).toArray();
    }
}

요렇게??돌려보니까 맞긴 하던데 확실히 바로바로 생각은 안나는것 같다...