Programmers/Lv.1

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

dbssk 2023. 3. 15. 17:33

문제 설명

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

제한사항

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

입출력 예

arrdivisorreturn

[5, 9, 7, 10] 5 [5, 10]
[2, 36, 1, 3] 1 [1, 2, 3, 36]
[3,2,6] 10 [-1]

내 풀이

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

class Solution {
    public int[] solution(int[] arr, int divisor) {
        ArrayList<Integer> list = new ArrayList<>();
        int[] answer = {};
        
        for (int num : arr) {
            if (num % divisor == 0) {
                list.add(num);
            }
        }
        
        if (list.size() != 0) {
            answer = new int[list.size()];
            int size = 0;
            for (int num : list) {
                answer[size++] = num;
            }
            Arrays.sort(answer);
        } else if (list.size() == 0) {
            answer = new int[1];
            answer[0] = -1;
        }
        
        return answer;
    }
}

풀이과정

  • 배열 값을 처음부터 초기화 할 수 없으므로 먼저 ArrayList에 divisor로 나누어 떨어지는 숫자를 추가한다.
  • list의 크기가 0이 아니라면 list의 크기만큼 배열을 만들고 값을 넣어준다.
  • 이후 Arrays.sort()를 이용해 오름차순으로 정렬해준다.
  • 만약 list의 크기가 0이라면 배열을 1크기로 만들고, -1을 값으로 넣어준다.
  • 시간 복잡도: O(n log n)