[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.*;
class Solution {
    public int[] solution(int[] arr, int divisor) {
        int[] answer = {};
        ArrayList<Integer> a1 = new ArrayList<Integer>();
        
        for(int i = 0; i <arr.length; i++){
            if(arr[i]%divisor == 0){
                a1.add(arr[i]);
            }
        }
        
        if(a1.isEmpty()){
            a1.add(-1);
        }
        
        answer = new int[a1.size()];
        
        for(int i = 0; i<a1.size() ; i++){
            answer[i] = a1.get(i);
        }
        
        Arrays.sort(answer);
        return answer;
    }
}

 

 

알고리즘 풀이 : 

1. int[] 타입으로 반환할 answer 변수를 초기화하여 선언

2. ArrayList인 a1을 Integer 타입으로 선언

3. for 문을 사용하여 arr 배열에 있는 원소를 divisor로 나누었을 때 나머지가 0인 원소만 add를 사용하여 a1에 추가

4. a1이 비어있을 경우 a1에 add를 이용하여 -1을 추가

- a1이 비어있다는 말은 arr 배열에 있는 모든 원소가 divisor로 나누었을 때 나머지가 0이 아닐 경우를 뜻함

5. answer의 크기를 a1의 size 만큼으로 설정

6. for문을 사용하여 a1에 있는 원소를 get으로 꺼내서 answer에 저장

7. answer을 Arrays.sort를 이용해 정렬

8. return을 사용하여 answer을 반환