프로그래머스 - 나누어 떨어지는 숫자 배열 / JAVA (자바) 코딩 알고리즘 연습

반응형

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr, int divisor) {
        ArrayList<Integer> result = new ArrayList<>();

		for (int i = 0; i < arr.length; i++) {
			if (arr[i] % divisor == 0) {
				result.add(arr[i]);
			}
		}
		
		int[] answer = {};
		
		if(result.size() != 0) {
			answer = new int[result.size()];
		}else {
            answer = new int[1];
			answer[0] = -1;
		}
		int idx = 0;
		
		for (int i : result) {
			answer[idx] = i;
			idx++;
		}
        
        Arrays.sort(answer);
        
        return answer;
    }
}