[프로그래머스] Java :: n의 배수 고르기

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/120905

프로그래머스

문제 설명

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

입출력 예시

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. 매개변수 "n"의 배수를 numlist의 인덱스에 찾아서 리턴하는 문제이다.

2. ArrayList를 하나 생성해서 numlist에 있는 값 중 n의 배수만 list에 넣어준다.

3. list에 넣어준 후 int 배열로 변환하여 리턴해준다.

코드

import java.util.*;
import java.util.stream.*;

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