[프로그래머스] 2차원으로 만들기 - c언어

문제 설명

정수 배열 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합니다.

소스코드

Solution.c

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

int** solution(int num_list[], size_t num_list_len, int n) {
    int row = num_list_len/n;
    int x = 0, y = 0;
    
    int** answer = (int**)malloc(sizeof(int*) * row);

    for(int i = 0; i < row; i++)
        answer[i] = (int *)malloc(sizeof(int) * n);

    for(int i = 0; i < num_list_len; i++){
        answer[x][y++] = num_list[i];
        if(y == n){
            y = 0;
            x++;
        }
    }

    return answer;
}

 

풀이

이번 문제의 핵심은 아래와 같은 2차원 배열의 동적할당이 핵심인 것 같다.

int** answer = (int**)malloc(sizeof(int*) * row);
answer[i] = (int *)malloc(sizeof(int) * n);

동적할당을 하고 나면 반복문을 이용하여 조건에 맞게 값을 배열 answer에 넣어주면 된다.

 

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

코딩테스트 연습 | 프로그래머스 스쿨