[프로그래머스Lv1] 나누어 떨어지는 숫자 배열

🚩 문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

📝 해결방법

answer 배열길이가 조건에 맞는 값만 들어가게 되는 가변형 길이라서 ArrayList를 이용해서 해결해본다.

📝 배운것

ArrayList
컬렉션 프레임웍에서 가장 많이 사용되는 컬렉션 클래스
순서o, 중복o

• 생성 : ArrayList list = new ArrayList<>();
• 추가 list.add(a);
• 가져오기 list.get(a);
• 삭제 list.remove(a);

👩🏻‍💻 내코드

import java.util.*;

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

📝 개선사항

출처 : programmers 나누어 떨어지는 숫자 배열