ALGORITHM/PROGRAMMERS

[프로그래머스] 나누어 떨어지는 숫자 배열 - JavaScript

thisisyjin 2022. 7. 20. 14:02

javascript

문제 설명

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]

입출력 예 설명

입출력 예#1
arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.

입출력 예#2
arr의 모든 원소는 1으로 나누어 떨어집니다. 원소를 오름차순으로 정렬해 [1, 2, 3, 36]을 리턴합니다.

입출력 예#3
3, 2, 6은 10으로 나누어 떨어지지 않습니다. 나누어 떨어지는 원소가 없으므로 [-1]을 리턴합니다.


내가 작성한 코드

function solution(arr, divisor) {
    let answer = [];
    arr.forEach(x => {
        if(x % divisor === 0) answer.push(x);
    })

    if (answer.length === 0) return [-1];
    return answer.sort((x,y) => x - y);
}

만약 divisor로 나눠떨어지면 push하고, 오름차순으로 정렬한 배열을 리턴한다.
length가 0이면 -1을 리턴한다.
forEach 말고 for..or를 사용할 수도 있다.

실행 결과

테스트 1 〉    통과 (0.10ms, 30.2MB)
테스트 2 〉    통과 (0.08ms, 30.2MB)
테스트 3 〉    통과 (0.08ms, 30MB)
테스트 4 〉    통과 (0.08ms, 30.2MB)
테스트 5 〉    통과 (0.10ms, 30.4MB)
테스트 6 〉    통과 (25.57ms, 33.5MB)
테스트 7 〉    통과 (0.18ms, 29.9MB)
테스트 8 〉    통과 (0.08ms, 29.9MB)
테스트 9 〉    통과 (0.14ms, 30MB)
테스트 10 〉    통과 (0.17ms, 30.2MB)
테스트 11 〉    통과 (0.11ms, 30.1MB)
테스트 12 〉    통과 (0.17ms, 29.9MB)
테스트 13 〉    통과 (0.24ms, 30.2MB)
테스트 14 〉    통과 (0.20ms, 30.3MB)
테스트 15 〉    통과 (0.17ms, 30.3MB)
테스트 16 〉    통과 (0.10ms, 30.1MB)

**

✅ 리팩토링

function solution(arr, divisor) {

    const answer = arr.filter(x => x % divisor === 0).sort((a,b) => a-b);
    return answer.length === 0 ? [-1] : answer;
}

filter 메서드를 이용한 버전이다.

실행 결과


테스트 1 〉    통과 (0.06ms, 29.9MB)
테스트 2 〉    통과 (0.08ms, 29.9MB)
테스트 3 〉    통과 (0.18ms, 30.5MB)
테스트 4 〉    통과 (0.07ms, 30.2MB)
테스트 5 〉    통과 (0.06ms, 30.4MB)
테스트 6 〉    통과 (21.16ms, 33.5MB)
테스트 7 〉    통과 (0.18ms, 30.2MB)
테스트 8 〉    통과 (0.06ms, 29.9MB)
테스트 9 〉    통과 (0.13ms, 30.1MB)
테스트 10 〉    통과 (0.16ms, 30.3MB)
테스트 11 〉    통과 (0.10ms, 29.9MB)
테스트 12 〉    통과 (0.11ms, 30MB)
테스트 13 〉    통과 (0.26ms, 30.2MB)
테스트 14 〉    통과 (0.16ms, 30.3MB)
테스트 15 〉    통과 (0.15ms, 30.4MB)
테스트 16 〉    통과 (0.09ms, 30MB)