[Programmers] 나누어 떨어지는 숫자 배열

programmers.co.kr/learn/courses/30/lessons/12910

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

 

Java

 

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

public class Solution {
	public int[] solution(int[] arr, int divisor) {
		
		List<Integer> tempList = new ArrayList<>();
		
		//나누어 떨어지는 수 tempList에 저장
		for(int i = 0; i < arr.length; ++i) {
			if(arr[i] % divisor == 0) {
				tempList.add(arr[i]);
			}
		}
		
		//나누어 떨어지는 element 없을 때 -1 담아 리턴
		if(tempList.size() == 0) {
			int[] answer = new int[] {-1};
			return answer;
		} else {
		//tempList 값을 answer에 저장, 정렬 후 리턴
			int[] answer = new int[tempList.size()];
			for(int i = 0; i < tempList.size(); ++i) {
				answer[i] = tempList.get(i);
			}
			Arrays.sort(answer);
			return answer;
		}

		
	}
}