《Cracking the Coding Interview》——第13章：C和C++——题目10

2014-04-25 20:47

题目：分配一个二维数组，尽量减少malloc和free的使用次数，要求能用a[i][j]的方式访问数据。

解法：有篇文章讲了六种new delete二维数组的方式，其中最后一种灰常高效。链接在此，解法六是巧妙的，不过里面的说法不对，而且还不标明转载原地址，可见这些技术网站的小编既不懂编程，也不尊重知识产权。没准这篇文章已经转载了无数次了。用这种方法创建和释放一个N维数组，只需要new和delete N次。我说的是N维，不是说长度。至于malloc和new的区别，就是C和C++的区别了。

代码：

// 13.10 Write a function in C called my2DAlloc, which allocates space for 2d array. Try to minimize the number of mallocs and frees.
#include <cstdio>
#include <cstdlib>
using namespace std;

int **my2DAlloc(int row, int col)
{
if (row < 1 || col < 1) {
return nullptr;
}

int i;
int **a;

a = (int **)malloc(row *  sizeof(int *));
a[0] = (int *)malloc(row * col * sizeof(int));
for (i = 1; i < row; ++i) {
a[i] = a[0] + i * col;
}

return a;
}

void my2DFree(int **ptr)
{
if (ptr == nullptr) {
return;
}
free(ptr[0]);
free(ptr);
}

int main()
{
int row, col;

scanf("%d%d", &row, &col);
int **a = my2DAlloc(row, col);

int i, j;

for (i = 0; i < row; ++i) {
for (j = 0; j < col; ++j) {
scanf("%d", &a[i][j]);
}
}

for (i = 0; i < row; ++i) {
for (j = 0; j < col; ++j) {
printf("%d ", a[i][j]);
}
printf("\n");
}

my2DFree(a);

return 0;
}