如果 malloc 失败，C 释放二维数组

Question

如果我有一个二维数组分配如下：

int** map;
map = malloc(number * sizeof(int*));
if(!(map)){
    printf("out of memory!\n");
    return 1;
}
for (int i = 0; i < number; i++){
    map[i] = malloc(number * sizeof(int));
    if (!(map[i])){
        printf("Not enough memory!\n");
        return 1;
    }
}

如果分配失败并且我们在 if 语句中输入，我是否应该释放映射和分配到现在的 "columns"？如果可以，我应该怎么做？

现在我只是打印消息和 return 1 但我不确定这是否是正确的方法。

Answer 1

您可以使用：

if (!(map[i])){
    printf("Not enough memory!\n");

    while (--i>=0)
        free(map[i]);

    free(map);
    return 1;
}

Answer 2

是的，你应该 free() 否则你会泄漏内存，如果这是在长运行ning 程序中，这可能很重要。

一种更简单的方法是计算所有分配的总大小，并进行单个较大的 malloc() 而不是一大堆较小的分配。这也（可能）更快，因为堆分配可能很昂贵。

这样的话，成功失败只需要检查一次，失败了就free()无所谓

像这样：

int ** map_allocate(size_t number)
{
  int **base = malloc(number * sizeof (int *) + number * number * sizeof (int));
  if(base != NULL)
  {
    int *row = (int *) (base + number);
    for(size_t i = 0; i < number; ++i)
      base[i] = row + i * number;
  }
  return base;
}

我没有测试-运行这个，但是类似的东西。

Answer 3

总是在分配指针时立即将指针设置为 null。

int **map;

map = malloc(number * sizeof(int *));
if(!map)
    goto out_of_memory;
for(i=0;i<number;i++)
    map[i] = 0;

for(i=0;i<numbers;i++)
{
    map[i] = malloc(number * sizeof(int));
    if(!map[i])
        goto out_of_memory;
}

...

return 0;

out_of_memory:
    if(map)
        for(i=0;i<number;i++)
            free(map[i]);
    free(map);
    return -1;

如果 malloc 失败，C 释放二维数组

C freeing 2D array if malloc fail

c

arrays

malloc

free