[프로그래머스 코딩테스트] 나누어 떨어지는 숫자 배열(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

import java.util.Arrays;
class Solution {
    public int[] solution(int[] arr, int divisor) {
        int count = 0;
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] % divisor == 0) {
                count++;
            }
        }
        if (count == 0) {
            return new int[] {-1};
        }
        
        int[] answer = new int[count];
        int index = 0;
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] % divisor == 0) {
                answer[index++] = arr[i];
            }
        }
        Arrays.sort(answer);
        return answer;
    }
}​

  answer에 딱맞는 배열크기를 생성해주기 위해 원소의 수를 셀 for문을 만들었습니다. count 변수를 선언합니다. for문이 arr의 길이만큼 순회합니다. 조건에 해당하는, 배열의 원소가 divisor로 나누어 떨어지는 경우 count가 1씩 증가합니다.

  만약 원소가 하나도 없다면 -1을 반환해야 하므로 if문을 통해 조건을 만들어주었습니다. new int[] {-1}은 원소로 -1을 갖는 크기 1의 int 배열을 생성한다는 뜻입니다.

  이후 answer 배열을 생성하고, 배열의 크기를 count로 선언했습니다. answer의 index가 될 index 변수도 생성했습니다. 앞서 작성했듯, 다시 원소를 구하는 것의 반복입니다. 조건에 부합할 시, answer[index]에 arr[i]가 할당될 것입니다. 할당될 때마다 index가 증가하므로 순서대로 쌓입니다.

  이 문제는 오름차순으로 정렬한 배열을 반환하기를 원합니다. 따라서 Arrays 클래스의 sort 메소드를 사용했습니다. 처음부터 정렬을 해도 무방합니다. 그러나 문제의 조건에 부합하지 않는 원소의 정렬은 필요하지 않으므로 조건문으로 걸러준 뒤 마지막에 정렬하는 것이 효율적일 것입니다.