정수 배열 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차원 배열로 변경합니다.

num_list n result
[1, 2, 3, 4, 5, 6, 7, 8] 2 [[1, 2], [3, 4], [5, 6], [7, 8]]

파이썬 첫번째

def solution(num_list, n):
    answer = []
    count = 0

    for i in range(len(num_list)//n):  
        answer.append(num_list[n*i:n*(i+1)])

    return answer

    

파이썬 두 번째

def solution(num_list, n):

    return [num_list[i-n:i] for i in range(n, len(num_list)+1, n)]

자바 첫번째

class Solution {
    public int[][] solution(int[] num_list, int n) {
        int[][] answer = {};
        int length = num_list.length;
        answer = new int[length/n][n];

        for(int i=0; i<length; i++){
            answer[i/n][i%n] = num_list[i];
        }

        return answer;
    }
} 

자바 두번째

import java.util.Arrays;
class Solution {
    public int[][] solution(int[] num_list, int n) {
        int[][] answer = new int[num_list.length/n][n];
        for(int i=0; i<num_list.length; i++){
            answer[i/n][i%n] = num_list[i];
        } 
        return answer;
    }
}