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

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

 

 

예제:

 

Input:
arr : [5, 9, 7, 10], divisor : 5
Output:
[5, 10]

Input:
arr : [3, 2, 6], divisor : 10
Output:
[-1]

 

제한 조건:

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

 

 

 

---

 

Solution #1

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

class Solution {
    public int[] solution(int[] arr, int divisor) {
        List<Integer> list = new ArrayList<>();

        for (int a : arr) {
            if (a % divisor == 0) list.add(a);
        }

        if (list.size() == 0) return new int[]{-1};

        int[] result = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }
        Arrays.sort(result);
                
        return result;
    }
}

 

Result #1

 

💡 조건에 맞는 배열 값을 ArrayList로 저장한 다음, 정렬 후 배열 형태로 리턴하였다.

 

---

 

Solution #2

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr, int divisor) {
        int[] result = Arrays.stream(arr).sorted().filter(i -> i % divisor == 0).toArray();
        if (result.length == 0) return new int[]{-1};
        return result;
    }
}

 

Result #2

 

💡 stream api를 사용하면 코드가 간결해진다. 하지만 실행 시간은..

 

 

---

 

More Algorithm!

 

👇👇

 

github.com/ggujangi/ggu.programmers

ggujangi/ggu.programmers

 

 

 

출처 : 프로그래머스