[프로그래머스 - JAVA] n의 배수 고르기

 

• 내 풀이

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

answer의 배열 크기를 초기화 시켜줘야하는데 배수의 값을 따로 구해서 초기화 시켜주고 다시 for문을 돌려서 answer에 넣기에는 코드를 반복해야하다보니 Arrays.stream을 이용하였다.

코딩테스트를 하면서 자주 사용하게 되었는데 매우 유용!! 

 

int[] answer의 타입을 ArrayList로 변경해서 add해도 되지만 문제의 틀을 깨지 않는 범위에서 코드를 작성해본다.