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

문제

정수 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합니다.

소스코드

class Solution {
    public int[] solution(int n, int[] numlist) {
        int num = 0; // n의 배수 구한 길이
        int idx = 0; // answer의 인덱스
        
        for(int i = 0; i < numlist.length; i++) {
            if(numlist[i] % n == 0)
            num++;
        }
        
        int[] answer = new int[num];
        for(int i = 0; i < numlist.length; i++) {
            if(numlist[i] % n == 0) {
                answer[idx] = numlist[i];
                idx++;
            }
        }
        
        return answer;
    }
}

 

풀이

num에는 n의 배수만큼의 길이를 구하여 넣어준 뒤, answer의 길이를 num으로 잡아준다. 두 번째 for문에서 한 번 더 배수를 구해준 뒤, answer[idx]에 numlist[i]를 넣어주면 n의 배수 값만 idx에 들어간다. 여기서, idx++를 해주지 않는다면 answer의 index가 증가하지 않아서 값이 한 개만 들어간다.