如果malloc失败,C释放2D数组

Question

如果我有一个如下分配的2D数组:

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语句我应该释放map和直到现在分配的"列"吗？如果是这样,我该怎么办？

现在我只打印消息并返回1,但我不确定这是否是正确的方法.

Answer 1

是的,free()否则你应该泄漏内存,如果这是在一个长期运行的程序中可能很重要.

使其更容易的一种方法是计算所有分配的总大小,并做一个较大的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;
}

我没有测试这个,但是沿着这些方向.