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语句我应该释放地图和分配到目前为止的“列”吗？如果是这样，我该怎么做？

现在我只打印消息并返回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;
}

我没有测试过这个，而是沿着这些方向进行测试。

Answer 2

您可以使用：

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

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

    free(map);
    return 1;
}

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释放2D数组

3 个答案: