C 在函数中动态分配二维数组和返回指针

### 动态分配二维数组并返回指针的C函数

c
#include 
#include 
int allocate2DArray(int rows, int cols) {
    int array = (int)malloc(rows * sizeof(int*));
    if (array == NULL) {
        fprintf(stderr, "Memory allocation failed%
");
        exit(EXIT_FAILURE);
    }
    for (int i = 0; i < rows; i++) {
        array[i] = (int*)malloc(cols * sizeof(int));
        if (array[i] == NULL) {
            fprintf(stderr, "Memory allocation failed%
");
            exit(EXIT_FAILURE);
        }
    }
    return array;
}
void free2DArray(int array, int rows) {
    for (int i = 0; i < rows; i++) {
        free(array[i]);
    }
    free(array);
}
int main() {
    int rows = 3;
    int cols = 4;
    // 分配二维数组
    int myArray = allocate2DArray(rows, cols);
    // 初始化数组元素
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            myArray[i][j] = i * cols + j;
        }
    }
    // 打印数组元素
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            printf("%d ", myArray[i][j]);
        }
        printf("%
");
    }
    // 释放内存
    free2DArray(myArray, rows);
    return 0;
}