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

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

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

 

입출력 예

arr	divisor	return
[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.Arrays;

class Solution {
    public int[] solution(int[] arr, int divisor) {
        int[] answer = {};
        int[] tmp = new int[arr.length];
        int cnt = 0;
        
        for(int num : arr) {
        	if(num % divisor == 0) {
        		tmp[cnt++] = num;
        	}
        }
        
        if(cnt != 0) {
        	answer = new int[cnt];
        	for(int i=0; i<cnt; i++) {
        		if(tmp[i] != 0) answer[i] = tmp[i];
        		System.out.println("i : " + i + " - " + answer[i]);
        	}
        	Arrays.sort(answer);
        } else {
        	answer = new int[1];
        	answer[0] = -1;
        }
        return answer;
    }
}

 

---

로컬 테스트 코드

public static int[] solution(int[] arr, int divisor) {
    int[] answer = {};
    int[] tmp = new int[arr.length]; // 나누어 떨어지는 값 저장 - 크기는 arr길이만큼
    int cnt = 0; // divisor로 나누어 떨어지는 개수
    
    for(int num : arr) { // arr 원소 하나씩 num에 저장
    	if(num % divisor == 0) { // 해당 원소가 divisor로 나누어떨어지는지 확인
    		tmp[cnt++] = num; // 나누어떨어진다면 해당값 cnt 인덱스에 저장 후 1 증가 
    	}
    }
    
    if(cnt != 0) { // 나누어 떨어지는값이 있다면
    	answer = new int[cnt]; // 해당 길이만큼 answer에 크기 할당
    	for(int i=0; i<cnt; i++) {
    		if(tmp[i] != 0) answer[i] = tmp[i];
    		System.out.println("i : " + i + " - " + answer[i]);
    	}
    	Arrays.sort(answer); // answer에 원소값들을 오름차순으로 정렬
    } else { // 나누어 떨어지는값이 없다면
    	answer = new int[1]; // 1크기로 초기화 후
    	answer[0] = -1; // -1 저장
    }
    return answer;
}