[Lv 1] 나누어 떨어지는 숫자 배열 / JAVA

문제

[ 문제 설명 ]

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

 

[ 제한 사항 ]

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

 

 

 

 

풀이

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


class Solution {
    public int[] solution(int[] arr, int divisor) {
    	// 크기지정필요없는 ArrayList에 넣자
    	ArrayList<Integer> list = new ArrayList<Integer>();
    	// 조건에 맞게 하나씩 넣어주자
        for (int i = 0; i < arr.length; i++) {
        	if(arr[i]%divisor==0) {
        		list.add(arr[i]);
        	}
		}
        // 조건에 맞는게 없었으면 -1 출력
        if(list.size()==0) {list.add(-1);}
        
        // 담긴 ArrayList크기만큼의 배열 선언
        int[] answer = new int[list.size()];
       
        for (int i = 0; i < list.size(); i++) {
			answer[i] = list.get(i);
		}
        
        // 정렬
        Arrays.sort(answer);
        
        return answer;
    }
}

 

array안의 각 값을 divisor로 나눴을때 0이 되면 list에 그 값을 add해준다.

add한 값이 없다면 즉, list의 size가 0이라면 list에 -1을 add해주고

배열 answer에 넣어 정렬해준 후 리턴한다.