[프로그래머스] 배열의 원소만큼 추가하기 (Javascript)

문제

프로그래머스

요구사항
아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 양의 정수 배열 arr가 매개변수로 주어질 때, 
arr의 앞에서부터 차례대로 원소를 보면서 원소가 a라면 X의 맨 뒤에 a를 a번 추가하는 일을 반복한 뒤의 배열 X를 return 하는 solution 함수를 작성해 주세요.

제한 조건
- 1 ≤ arr의 길이 ≤ 1001 ≤
- arr의 원소 ≤ 100

입출력

arr	result
[5, 1, 4]	[5, 5, 5, 5, 5, 1, 4, ,4, 4, 4]
[6, 6]	[6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]
[1]	[1]

 

 

내가 생각한 로직

1. forEach 메서드를 이용하여 배열의 요소에 접근한다.

2. for 문을 이용하여 요소의 크기만큼 빈 배열에 해당 요소를 넣는다.

 

forEach 안에서 for 반복문을 또 쓰고 싶지 않았지만,

해당 요소만큼 배열 안에 넣는 방법이 for문 밖에 떠오르지 않았다.

 

 

내가 구현한 코드

function solution(arr) {
    let x = [];
    arr.forEach(a => {
        for (let i = 0; i < a; i++) {
            x.push(a);
        }
    })
    return x;
}

 

 

다른 해결 방법

사용한 도구: reduce(), fill(). new Array()

function solution(arr) {
    return arr.reduce((list, num) => [...list, ...new Array(num).fill(num)], []);
}

reduce 함수를 이용하여 값이 누적되도록 하였다.

arr이 [5, 1, 4]라고 해보자.

new Array(5)를 통해 배열의 길이가 5인 빈 배열을 만들었다.

fill(5)를 통해 빈 배열에 5로 가득 채웠다.

[...list, ...new Array(5).fill(5)]을 통해 기존에 있던 배열이 누적되도록 만든다.