函数动态分配矩阵并处理所有错误

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

int** allocateMatrix(int rows, int cols) {
    int** matrix = (int**)malloc(rows * sizeof(int*));
    if (matrix == NULL) {
        printf("Failed to allocate memory for matrix.\n");
        return NULL;
    }

    for (int i = 0; i < rows; i++) {
        matrix[i] = (int*)malloc(cols * sizeof(int));
        if (matrix[i] == NULL) {
            printf("Failed to allocate memory for matrix row %d.\n", i);
            // 释放已分配的内存
            for (int j = 0; j < i; j++) {
                free(matrix[j]);
            }
            free(matrix);
            return NULL;
        }
    }

    return matrix;
}

void freeMatrix(int** matrix, int rows) {
    if (matrix == NULL) {
        return;
    }

    for (int i = 0; i < rows; i++) {
        free(matrix[i]);
    }
    free(matrix);
}

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

    int** matrix = allocateMatrix(rows, cols);
    if (matrix == NULL) {
        return 1;
    }

    // 使用矩阵进行计算或其他操作

    freeMatrix(matrix, rows);

    return 0;
}