动态分配三维数组的C函数(使用malloc)

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

int*** allocate3DArray(int rows, int cols, int depth) {
    int*** arr = (int***)malloc(rows * sizeof(int**));
    for (int i = 0; i < rows; i++) {
        arr[i] = (int**)malloc(cols * sizeof(int*));
        for (int j = 0; j < cols; j++) {
            arr[i][j] = (int*)malloc(depth * sizeof(int));
        }
    }
    return arr;
}

void deallocate3DArray(int*** arr, int rows, int cols, int depth) {
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            free(arr[i][j]);
        }
        free(arr[i]);
    }
    free(arr);
}

int main() {
    int rows = 3;
    int cols = 4;
    int depth = 5;

    int*** arr = allocate3DArray(rows, cols, depth);

    // 使用动态分配的三维数组进行操作
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            for (int k = 0; k < depth; k++) {
                arr[i][j][k] = i + j + k;
                printf("%d ", arr[i][j][k]);
            }
            printf("\n");
        }
        printf("\n");
    }

    deallocate3DArray(arr, rows, cols, depth);

    return 0;
}