关于C语言中使用malloc为二维数组分配内存的技术疑问

在C语言里，要实现返回整数二维数组指针的函数，通常有两种常用的内存分配方式，下面分别说明：

方法一：指针数组（每行独立分配） #

这种方式先分配一个存储指针的数组，再给每个指针分配对应行的内存，优点是每行长度可以不同，理解起来更直观。

示例代码 #

#include <stdlib.h>

// 返回指向指针的指针（即二维数组的指针）
int** allocate_2d_array(int rows, int cols) {
    // 第一步：分配存储每行指针的数组
    int** arr = (int**)malloc(rows * sizeof(int*));
    if (arr == NULL) {
        return NULL; // 内存分配失败
    }

    // 第二步：为每一行分配内存
    for (int i = 0; i < rows; i++) {
        arr[i] = (int*)malloc(cols * sizeof(int));
        if (arr[i] == NULL) {
            // 如果某一行分配失败，释放已分配的内存避免泄漏
            for (int j = 0; j < i; j++) {
                free(arr[j]);
            }
            free(arr);
            return NULL;
        }
    }

    return arr;
}

// 对应的内存释放函数
void free_2d_array(int** arr, int rows) {
    for (int i = 0; i < rows; i++) {
        free(arr[i]);
    }
    free(arr);
}

使用说明 #

• 调用allocate_2d_array(3, 4)会得到一个3行4列的二维数组，访问方式和普通二维数组一致：arr[i][j]
• 必须成对调用free_2d_array释放内存，否则会造成内存泄漏

方法二：连续内存块的二维数组 #

这种方式把整个二维数组作为一块连续的内存分配，内存利用率更高，缓存友好，适合固定行列数的场景。

方式A：用指向数组的指针 #

#include <stdlib.h>

// 返回指向int[cols]类型数组的指针
int (*allocate_contiguous_2d(int rows, int cols))[cols] {
    // 计算总内存大小：行数 × 每行元素数 × 每个int的大小
    int (*arr)[cols] = (int (*)[cols])malloc(rows * cols * sizeof(int));
    return arr;
}

// 释放内存
void free_contiguous_2d(int (*arr)[cols]) {
    free(arr);
}

方式B：用一维数组模拟二维访问（兼容旧编译器） #

如果编译器不支持变长数组（VLA），可以用这种方式，通过计算下标访问元素：

#include <stdlib.h>

int* allocate_flat_2d(int rows, int cols) {
    // 分配连续的内存块
    int* arr = (int*)malloc(rows * cols * sizeof(int));
    return arr;
}

// 访问第i行第j列的元素：*(arr + i*cols + j) 或者 arr[i*cols + j]
// 释放内存直接free(arr)即可

使用说明 #

• 连续内存的方式只需要一次free操作，比方法一更高效
• 方式A中，访问方式依然是arr[i][j]，和普通二维数组完全一致
• 方式B需要手动计算下标，兼容性更好但稍显繁琐

两种方法对比 #

• 指针数组：灵活支持每行不同长度，但内存碎片化，释放时需要逐行释放
• 连续内存块：内存连续，缓存效率高，释放简单，但行列数固定（方式A依赖VLA，方式B需手动计算下标）

内容的提问来源于stack exchange，提问作者Artisan