[코딩테스트 Lv.0] 2차원으로 만들기

정수 배열 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]]

 

제한사항

num_list의 길이는 n의 배 수개입니다.
0 ≤ num_list의 길이 ≤ 150
2 ≤ n < num_list의 길이

 

입출력 예

num_list				n	result
[1, 2, 3, 4, 5, 6, 7, 8]		2	[[1, 2], [3, 4], [5, 6], [7, 8]]
[100, 95, 2, 4, 5, 6, 18, 33, 948]	3	[[100, 95, 2], [4, 5, 6], [18, 33, 948]]

 

입출력 예 설명

입출력 예 #1
num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 2 * 4 배열로 변경한 [[1, 2], [3, 4], [5, 6], [7, 8]] 을 return합니다.

입출력 예 #2
num_list가 [100, 95, 2, 4, 5, 6, 18, 33, 948] 로 길이가 9이고 n이 3이므로 3 * 3 배열로 변경한 [[100, 95, 2], [4, 5, 6], [18, 33, 948]] 을 return합니다.

 

---

내 풀이

[Java]

class Solution {
    public int[][] solution(int[] num_list, int n) {
        int[][] answer = new int[num_list.length / n][n];
        int cnt = 0;
        for(int i = 0; i < answer.length; i++){
            for(int j = 0; j < n; j++){
                answer[i][j] = num_list[cnt++];
            }
        }
        
        return answer;
    }
}

 

[C]

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// num_list_len은 배열 num_list의 길이입니다.
int** solution(int num_list[], size_t num_list_len, int n) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int num_len = num_list_len / n;
    int** answer = (int**)malloc(sizeof(int*)* num_len);
    
    int a = 0, b = 0 ;
    
    for(int i = 0; i < num_len; i++){
        answer[i] = (int*)malloc(sizeof(int)* n);
    }
    
    for(int i = 0; i < num_list_len; i++){
        answer[a][b++] = num_list[i];
        if(b == n){
            b = 0;
            a++;
        }
    }
    return answer;
}

 

[Python]

def solution(num_list, n):
    answer = []
    
    for i in range(0, len(num_list), n):
        answer += [num_list[i : i + n]]
    return answer

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/120842

프로그래머스