c - 如何为二维数组动态分配连续的内存块

Question

如果我像这样分配一个二维数组 int a[N][N];它将分配一个连续的内存块。

但是如果我尝试像这样动态地做:

int **a = malloc(rows * sizeof(int*));
for(int i = 0; i < rows; i++) 
   a[i] = malloc(cols * sizeof(int));

这会在行中的元素之间保持一个单位步长，但行之间可能并非如此。

一种解决方案是将二维转换为一维，除此之外，还有其他方法吗？

Answer 1

如果您的数组维度在编译时已知:

#define ROWS ...
#define COLS ...

int (*arr)[COLS] = malloc(sizeof *arr * ROWS);
if (arr) 
{
  // do stuff with arr[i][j]
  free(arr);
}

如果您的数组维度在编译时未知，并且您使用的是支持可变长度数组的 C99 编译器或 C2011 编译器:

size_t rows, cols;
// assign rows and cols
int (*arr)[cols] = malloc(sizeof *arr * rows);
if (arr)
{
  // do stuff with arr[i][j]
  free(arr);
}

如果您的数组维度在编译时未知，并且您不使用支持可变长度数组的 C99 编译器或 C2011 编译器:

size_t rows, cols;
// assign rows and cols
int *arr = malloc(sizeof *arr * rows * cols);
{
  // do stuff with arr[i * rows + j]
  free(arr);
}