Algorithm/PROGRAMMERS[Java]

Lv1. 나누어 떨어지는 숫자 배열

씨니 2022. 4. 17. 11:36
728x90

▶ 문제

 

▶ 풀이방법

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int divisor) {
        int[] answer = {};
        
        LinkedList<Integer> list = new LinkedList<>();
        for(int a : arr) if(a % divisor == 0) list.add(a); //나눠지는 수 list에 저장.
        if(list.size() == 0){ //아무것도 없을 경우 배열에 -1 삽입
            answer = new int[1];
            answer[0] = -1;
        }else{
            answer = new int[list.size()];
            for(int i = 0; i < list.size(); i++){
                answer[i] = list.get(i); //list에 저장된 값 꺼내서 결과 배열에 저장.
            }
            Arrays.sort(answer);    
        }
        return answer;
    }
}

 

728x90