谁能解释一下如何从函数返回 C 中的二维数组？

Question

我是 C 新手，在学习过程中我想从函数返回一个二维数组，以便我可以在我的主程序中使用它。谁能用例子向我解释一下。提前致谢。

Answer 1

这取决于它如何实施。您可以仅使用一个一维数组，其中您知道每行（行）的长度，并且下一行紧接着上一行开始。或者，您可以有一个指向数组的指针数组。不过，额外的成本是您需要取消引用两个指针才能获取一个数据元素。

// 2D array of data, with just one array
char* get_2d_ex1(int rows, int cols) {
    int r, c, idx;
    char* p = malloc(rows*cols);

    for (r=0; r<rows; r++) {
        for (c=0; c<cols; c++) {
            idx = r*cols + c;   // this is key
            p[idx] = c;         // put the col# in its place, for example.
        }
     }
     return p;
}

Answer 2

将您的函数声明为返回一个指向指针的指针。如果我们使用 int 为例：

int **samplefunction() {
    int** retval = new int*[100];
    for (int i = 1; i < 100; i++) {
        retval[i] = new int[100];
    }
    // do stuff here to retval[i][j]
    return retval;
}

Answer 3

以下是如何创建、操作和释放“二维数组”的示例：

#include <stdlib.h>

#define ROWS   5
#define COLS   10

int **create_2d_array(size_t rows, size_t cols)
{
   size_t i;   
   int **array = (int**)malloc(rows * sizeof(int*));
   for (i = 0; i < rows; i++)
      array[i] = (int*)malloc(cols * sizeof(int));

   return array;
}

void free_2d_array(int **array, size_t rows, size_t cols)
{
   size_t i;
   for (i = 0; i < rows; i++)
      free(array[i]);

   free(array);
}

int main(void)
{
   int **array2d = create_2d_array(ROWS, COLS);
   /* ... */
   array2d[3][4] = 5;
   /* ... */
   free_2d_array(array2d, ROWS, COLS);

   return 0;
}

要创建“二维数组”/矩阵，您所要做的就是创建一个动态指针数组（在此示例中） case int*) 的行/宽度的大小：

int **array = (int**)malloc(rows * sizeof(int*));

然后将每个指针设置为指向列/高度的大小的 int 的动态数组:

array[i] = (int*)malloc(cols * sizeof(int));

请注意，强制转换malloc 不是必需的，这只是我的一个习惯。