careercup-C和C++ 13.10
13.10 用C编写一个my2DALLoc函数,可分配二维数组。将malloc函数的调用次数降到最少,并确保可通过arr[i][j]访问该内存。
解法:
这道题目最简单的方法就是先开一个数组来存储指向每一行的指针, 然后再为每一行动态地分配空间。这是非常常见的动态申请二维数组空间的方法:
int** My2DAlloc(int rows, int cols){ int **arr = (int**)malloc(rows*sizeof(int*)); for(int i=0; i<rows; ++i) arr[i] = (int*)malloc(cols*sizeof(int)); return arr; }
上述方法使用了(rows+1)次的malloc,malloc使用过多会影响程序的运行效率, 那么有没有办法减少malloc的使用呢。
虽然我们做的事情是动态申请二维数组空间,但这些申请的空间本质上是一维, 只不过有些空间存储了地址,而有些空间则存储了数据。比如上面的方法, 申请了一个长度为rows的一维数组,里面存放的是指针(int*),指向每一行的地址。 然后又申请了rows*cols大小的空间,里面存放的是整型数据(int)。既然如此, 我们一次性将这么多的空间申请下来,然后在该存放地址的空间存放地址, 在该存放数据的空间存放数据就OK了。
我们需要存储指向每一行的地址,大小为:
int header = rows * sizeof(int*);
同时需要存储rows*cols的整型数据,大小为:
int data = rows * cols * sizeof(int);
我们一次性将这些空间申请下来:
int **arr = (int**)malloc(header + data);
由于前面rows * sizeof(int*)的大小存放的是指针,因此arr类型是int**。 而跨过rows个单元后,后面存放的是整型数据,因此需要将其类型转为int*:
int *buf = (int*)(arr + rows);
最后,从buf指向的地址开始,每cols个单元组成一行,将行首地址存放到arr 的相应位置即可。
for(int i=0; i<rows; ++i) arr[i] = buf + i * cols;
代码如下:
int** My2DAlloc1(int rows, int cols){ int header = rows * sizeof(int*); int data = rows * cols * sizeof(int); int **arr = (int**)malloc(header + data); int *buf = (int*)(arr + rows); for(int i=0; i<rows; ++i) arr[i] = buf + i * cols; return arr; }
这样一来,我们使用一次的malloc就可以动态地申请二维数组空间, 并且可以用arr[i][j]对数组元素进行访问。