[Programmers] 나누어 떨어지는 숫자 배열 - Java

문제

프로그래머스

 

풀이

배열에 있는 수에서 divisor로 나누어 떨어지는 값을 오름차순으로 정렬된 값을 구하는 문제이다.

 

1. Arrays.stream()을 이용해서 divisor로 나누어 떨어지는 값을 추출하고 오름차순을 정렬한다.

 

2. 배열이 비어있을 때 예외처리를 해준다.

 

3. 배열을 리턴한다.

 

소스코드

import java.util.Arrays;

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

 

실행결과