코딩 테스트

프로그래머스 나누어 떨어지는 숫자 배열 Java (+1)

hyecozy 2022. 11. 15. 16:12

https://school.programmers.co.kr/learn/courses/30/lessons/12910

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

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

class Solution {
    public int[] solution(int[] arr, int divisor) {
        
        List<Integer> answerList = new ArrayList<>();
        
        for(int i = 0; i < arr.length; i++){
            if(arr[i] % divisor == 0){
                answerList.add(arr[i]);        
            }
        }
        
        if(answerList.size() == 0){
            answerList.add(-1);
        }
        
        int[] answer = answerList.stream()
                        .mapToInt(Integer::intValue)
                        .toArray();
        
        Arrays.sort(answer);
        return answer;
    }
}

결과값의 길이가 정해져 있지 않아서 List를 쓰는 게 좋다고 생각했다

준비된 배열도 있으니 for문을 쓸 때 향상된 for문을 썼으면 더 좋았을 거 같다

 

저번에 푼 문제에서 알게 된 정렬 메소드를 쓸 수 있어서 좋았다