pwhTxU - Online C Compiler & Debugging Tool

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

int main(){
    int i, j;
    int a, b;
    int **mat;
    scanf("%d %d", &a, &b);

    // 2次元配列の動的確保
    mat = (int **)malloc(a * sizeof(int *));
    if(mat == NULL){
        printf("Memory allocation failed\n");
        return 1;
    }
    for(i = 0; i < a; i++){
        mat[i] = (int *)malloc(b * sizeof(int));
        if(mat[i] == NULL){
            printf("Memory allocation failed\n");
            return 1;
        }
    }

    // 数値を代入
    int k = 1;
    for(i = 0; i < a; i++){
        for(j = 0; j < b; j++){
            mat[i][j] = k++;
        }
    }

    // 表示部分（既存コード）
    for(i = 0; i < a; i++){
        for(j = 0; j < b; j++){
            printf("%d ", mat[i][j]);
        }
        printf("\n");
    }

    // メモリ解放
    for(i = 0; i < a; i++){
        free(mat[i]);
    }
    free(mat);

    return 0;
}