[ 코딩 테스트 ] 나누어 떨어지는 숫자 배열

문제

조건

 

정답

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int[] solution(int[] arr, int divisor) {
        
        ArrayList<Integer> list = new ArrayList<Integer>();
        
        // 나누어 떨어질 때
        for (int num : arr) {
            if (num % divisor == 0) {
                list.add(num);
            }
        }
        
        // 나누어 떨어지지 않을 때
        if (list.size() == 0) {
            int[] answer = {-1};
            return answer;
        }
        
        // 오름차순
        Collections.sort(list);
        
        // ArrayList를 배열로
        int[] answer = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }
        
        return answer;
    }
}

 

---

 

후기

 

초기 코드인 배열로 하려다가...

헷갈려서 그냥 컬랙션으로 바꿔버렸다