프로그래머스(나누어 떨어지는 숫자 배열) - Java

[나누어 떨어지는 숫자 배열]

https://programmers.co.kr/learn/courses/30/lessons/12910

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

 

* 로직

• arr 배열을 오름차순 정렬한다
• divisor로 나누어 떨어지는 수를 담는다

 

import java.io.*;
import java.util.*;

class Solution {
    public int[] solution(int[] arr, int divisor) {
        int[] answer = {};
        Arrays.sort(arr);
        
        ArrayList<Integer> result = new ArrayList<>();
        for(int input : arr) {
            if(input % divisor == 0) {
                result.add(input);
            }
        }
                
        if(result.isEmpty()) {
            answer = new int[1];
            answer[0] = -1;
        }
        
        else {
            answer = new int[result.size()];
            for(int i=0; i<answer.length; ++i) {
                answer[i] = result.get(i);
            }  
        }
        
        return answer;
    }
}