[프로그래머스] 나누어 떨어지는 숫자 배열 - 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.*;

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

 

1. stream 의  filter()를 이용하여 나누어 떨어지는 수를 걸러내고,  sorted() 로 오름차순 정렬한 연산을 toArray()로  answer에 담는다.

2. answer에 길이가 0이면 나누어 떨어지는 수가 없다는 의미이므로 -1을 담아서 반환