从函数返回一个二维数组

Question

我是C的新手.我试图从一个函数返回一个二维数组.就是这样的

int *MakeGridOfCounts(int Grid[][6])
{
  int cGrid[6][6] = {{0, }, {0, }, {0, }, {0, }, {0, }, {0, }};
  int (*p)[6] = cGrid;
  return (int*)p;
}

我知道这会导致错误,需要帮助.谢谢

Answer 1

C语言有一个基本缺陷:不可能从函数返回数组.这有很多变通方法; 我将描述三个.

用指向数组的指针替换

返回指针而不是数组本身.这导致了C中的另一个问题:当一个函数返回指向某个东西的指针时,它通常应该动态地分配一些东西.你不应该忘记稍后解除分配(当不再需要数组时).

typedef int (*pointer_to_array)[6][6];

pointer_to_array workaround1()
{
    pointer_to_array result = malloc(sizeof(*result));
    (*result)[0][0] = 0;
    (*result)[1][0] = 0;
    (*result)[2][0] = 0;
    (*result)[3][0] = 0;
    (*result)[4][0] = 0;
    (*result)[5][0] = 0;
    return result;
}

用指向int的指针替换

二维数组在内存中显示为一系列数字,因此您可以通过指向第一个元素的指针替换它.您明确声明要返回一个数组,但是您的示例代码返回一个指向int的指针,因此您可以相应地更改其余代码.

int *workaround2()
{
    int temp[6][6] = {{0}}; // initializes a temporary array to zeros
    int *result = malloc(sizeof(int) * 6 * 6); // allocates a one-dimensional array
    memcpy(result, temp, sizeof(int) * 6 * 6); // copies stuff
    return result; // cannot return an array but can return a pointer!
}

用结构包裹

听起来很傻,但功能可以返回结构,即使它们不能返回数组!即使返回的结构包含数组.

struct array_inside
{
    int array[6][6];
};

struct array_inside workaround3()
{
    struct array_inside result = {{{0}}};
    return result;
}

Answer 2

试试这个，在我的 mac 上用 GCC 编译得很好。

typedef struct _intGrid_t{
    int **data;
    int width;
    int height;
} *IntGrid;

IntGrid makeGridWithSize(int width, int height);
IntGrid makeGridWithSize(int width, int height)
{
    IntGrid grid = malloc(sizeof(struct _intGrid_t));

    int **data = (int **) malloc(sizeof(int *) * width);

    for (int i = 0; i < width; i++) {
        data[i] = malloc(sizeof(int) * height);

        memset(data[i], 0, sizeof(int) * height);
    }

    grid->data = data;
    grid->width = width;
    grid->height = height;

    return grid;
}

void printGrid(IntGrid grid);
void printGrid(IntGrid grid)
{
    printf(" { \n");
    for (int i =0; i < grid->width; i++) {
        printf(" { ");

        for (int j = 0; j < grid->height; j++) {
            printf("%i", grid->data[i][j]);

            if (j != grid->height - 1)
            {
                printf(", ");
            }
        }

        printf(" } \n");
    }
    printf(" } \n");
}

void freeGrid(IntGrid grid);
void freeGrid(IntGrid grid)
{
    for (int i = 0; i < grid->width; i++) {
        free(grid->data[i]);
    }

    free(grid->data);
    free(grid);
}


int main (int argc, const char * argv[])
{
    srand((int) time(NULL));

    IntGrid grid = makeGridWithSize(10, 10);


    for (int i = 0; i < grid->width; i++) {
        for (int j = 0; j < grid->height; j++) {
            grid->data[i][j] = rand() % 10;
        }
    }

    printGrid(grid);
    freeGrid(grid);

    return 0;
}