코딩 테스트
프로그래머스 나누어 떨어지는 숫자 배열 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문을 썼으면 더 좋았을 거 같다
저번에 푼 문제에서 알게 된 정렬 메소드를 쓸 수 있어서 좋았다