프로그래머스/프로그래머스 - JAVA

[JAVA] 나누어 떨어지는 숫자 배열

메타듀 2023. 2. 23. 17:55
728x90
반응형

문제 설명

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]

코드

처음에는

 if(divisor == 1){
            Arrays.sort(arr);
            return arr;
        }
        List<Integer> list = new ArrayList();

        for (int i = 0; i < arr.length; i++) {
            if(arr[i]%divisor == 0){
                list.add(arr[i]);
            }
        }

        int listSize = list.size();
        Integer[] result = list.toArray(new Integer[listSize]);

        int[] answer = Arrays.stream(result).mapToInt(i -> i).toArray();

        if(answer.length == 0){
            answer = new int[1];
            answer[0] = -1;
        }
        return answer;

이렇게 짜보았다.. 코드실행에서는 성공했는데 제출 채점에서는 실패..!
근데 아직 이유는 모르겠다.. 연구해봐야지..
다른 방법을 생각해서

import java.util.Arrays;

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++;
            }
        }

        int[] answer = new int[count];
        count = 0;
        for (int i = 0; i < arr.length; i++) {
            if(arr[i]%divisor == 0){
                answer[count]=arr[i];
                count++;
            }
        }

        Arrays.sort(answer);

        if(answer.length == 0){
            answer = new int[1];
            answer[0] = -1;
        }
        return answer;
    }
}

이 코드로 성공!

반응형

'프로그래머스 > 프로그래머스 - JAVA' 카테고리의 다른 글

[JAVA] 예산  (0) 2023.02.23
[JAVA] 시저 암호  (0) 2023.02.23
[JAVA] 2016년  (0) 2023.02.23
[JAVA] 최대공약수와 최소공배수  (0) 2023.02.23
[JAVA] 신규 아이디 추천  (0) 2023.02.23