나누어 떨어지는 숫자 배열

문제 설명

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

제한사항

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

입출력 예

arrdivisorreturn

[5, 9, 7, 10]	5	[5, 10]
[2, 36, 1, 3]	1	[1, 2, 3, 36]
[3,2,6]	10	[-1]

입출력 예 설명

입출력 예#1
arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.

입출력 예#2
arr의 모든 원소는 1으로 나누어 떨어집니다. 원소를 오름차순으로 정렬해 [1, 2, 3, 36]을 리턴합니다.

입출력 예#3
3, 2, 6은 10으로 나누어 떨어지지 않습니다. 나누어 떨어지는 원소가 없으므로 [-1]을 리턴합니다.

 

//

 

import java.util.ArrayList;
import java.util.Collections;

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

조건에 맞는 요소를 담기 위한 ArrayList를 list로 인스턴스하고

 

for문 안에 있는 if문의 조건이 맞는다면 list에 add하도록 하였다.

 

다음으로 조건에 맞는 배열의 요소가 없다면 list의 size는 0이므로 return하게 해주었고

 

이후 Collections.sort를 이용하여 오름차순 정렬을 한 뒤 배열에 넣어 return하였다.