알고리즘 풀이/프로그래머스

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

배게 2019. 3. 10. 07:21
728x90

divisor로 나누어 떨어지는 숫자들 ans 리스트에 추가


ans리스트에 추가된 숫자가 0개일 경우에 -1값 저장


오름차순 정렬


리스트를 배열로 바꿔준 후 리턴



 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import java.util.*;

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