[programmers/Java] 나누어 떨어지는 숫자 배열 - A004

-  문제

문제 설명

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

 

제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

 

-  알고리즘

arr를 받아 숫자를 하나씩 dicisor로 나누어 계산했다.

ArrayList와 일반 배열을 같이 사용해야했어서 조금 헷갈렸다.

결과값이 배열로 return 되다보니 편안하게 코딩하기 힘들었다.

그래도 쉬운 문제였기에 금방 해결했다.

 

-  코드

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