알고리즘테스트/Programmers

[Lv.1_Java] 나누어 떨어지는 숫자 배열

코딩거인 2023. 6. 1. 13:54
728x90

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int divisor) {
        int count = 0;
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] % divisor == 0) {
                count++;
            }
        }
        
        if (count == 0) {
            return new int[] {-1};
        }
        
        int[] answer = new int[count];
        int index = 0;
        
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] % divisor == 0) {
                answer[index] = arr[i];
                index++;
            }
        }
        Arrays.sort(answer);
        return answer;
    }
}

우선 answer 의 배열을 초기화 해주기 위해서 길이를 구해야했다 만약 나누어 떨어지는게 없다면 바로 -1 을 리턴해주고!

나누어 떨어지는 값들은 다시 값을 지정해 주기 위해서 for반복문을 돌려서 조건에 합치되는 요소들만 배열에 넣어준다!

그후 오름차순 정리를 하여 리턴해준다!

728x90