[Programmers] 120842번 - 2차원 배열

Problem

문제 보기

정수 배열 num_list와 정수 n이 매개변수로 주어집니다. num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.

num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 num_list를 2 * 4 배열로 다음과 같이 변경합니다. 2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.

---

Solution

function solution(num_list, n) {
    let answer = new Array(num_list.length / n).fill(0).map(() => new Array(n));
    let num = 0;
    for (let i = 0; i < num_list.length / n; i++){
        for (let j = 0; j < n; j++){
            answer[i][j] = num_list[num++];
        }
    }
    return answer;
}

새로운 2차원 배열 생성 후에 값을 대입하는 부분을 이중 for문을 이용하여 복잡하게 풀었다..

아래와 같이 2차원 배열 생성할 때 splice()를 이용해서 한줄로 만들 수 있다.

function solution(num_list, n) {
    return Array(num_list.length / n).fill([]).map(() => num_list.splice(0, n))
}